Skip to content

Commit 420416f

Browse files
committed
Merge branch 'install-fix' into 'r1.6.1'
set --noinstall=true for bare-metal, --noinstall=false for docker See merge request intelai/models!182
2 parents c4ea2a5 + 47c0e7f commit 420416f

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

benchmarks/launch_benchmark.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)