@@ -91,6 +91,12 @@ def parse_args(self):
9191 "--debug" , help = "Launches debug mode which doesn't execute "
9292 "start.sh when running in a docker container." , action = "store_true" )
9393
94+ arg_parser .add_argument (
95+ "--noinstall" ,
96+ help = "whether to install packages for a given model when running in docker "
97+ "(default --noinstall='False') or on bare metal (default --noinstall='True')" ,
98+ dest = "noinstall" , action = "store_true" , default = None )
99+
94100 return arg_parser .parse_known_args ()
95101
96102 def validate_args (self ):
@@ -192,6 +198,7 @@ def get_env_vars(self, benchmark_scripts, use_case, intelai_models,
192198 "FRAMEWORK" : args .framework ,
193199 "NUM_CORES" : args .num_cores ,
194200 "NUM_INTER_THREADS" : args .num_inter_threads ,
201+ "NOINSTALL" : str (args .noinstall ) if args .noinstall is not None else "True" if not args .docker_image else "False" ,
195202 "NUM_INTRA_THREADS" : args .num_intra_threads ,
196203 "DATA_NUM_INTER_THREADS" : args .data_num_inter_threads ,
197204 "NUM_TRAIN_STEPS" : args .num_train_steps ,
@@ -217,11 +224,6 @@ def get_env_vars(self, benchmark_scripts, use_case, intelai_models,
217224 split_arg [0 ] = split_arg [0 ].replace ("-" , "_" ).lstrip ('_' )
218225 env_var_dict [split_arg [0 ]] = split_arg [1 ]
219226
220-
221- # Set the default value for NOINSTALL, if it's not explicitly set by the user
222- if "NOINSTALL" not in env_var_dict :
223- env_var_dict ["NOINSTALL" ] = "False"
224-
225227 return env_var_dict
226228
227229 def run_bare_metal (self , benchmark_scripts , intelai_models ,
0 commit comments