diff --git a/script/app-mlperf-inference-mlcommons-python/customize.py b/script/app-mlperf-inference-mlcommons-python/customize.py index c048e8f3c..55e89b68a 100644 --- a/script/app-mlperf-inference-mlcommons-python/customize.py +++ b/script/app-mlperf-inference-mlcommons-python/customize.py @@ -250,6 +250,7 @@ def get_run_cmd_reference( env['OUTPUT_DIR'] = env['MLC_MLPERF_OUTPUT_DIR'] if env.get('MLC_MLPERF_VISION_DATASET_OPTION', '') == '' and env.get( 'MLC_MLPERF_DEVICE') != "tpu": + if os_info['platform'] == 'windows': cmd = "python python/main.py --profile " + env['MLC_MODEL'] + "-" + env['MLC_MLPERF_BACKEND'] + \ " --model=\"" + env['MLC_ML_MODEL_FILE_WITH_PATH'] + '" --dataset-path="' + env['MLC_DATASET_PREPROCESSED_PATH'] + \ @@ -261,6 +262,8 @@ def get_run_cmd_reference( cmd = "./run_local.sh " + env['MLC_MLPERF_BACKEND'] + ' ' + \ env['MLC_MODEL'] + ' ' + env['MLC_MLPERF_DEVICE'] + " --scenario " + env['MLC_MLPERF_LOADGEN_SCENARIO'] + " " + env['MLC_MLPERF_LOADGEN_EXTRA_OPTIONS'] + \ scenario_extra_options + mode_extra_options + dataset_options + if env['MLC_MODEL'] in ["resnet50"]: + cmd += f" --max-batchsize {env.get('MLC_MLPERF_LOADGEN_MAX_BATCHSIZE', '128')} " return cmd, env['RUN_DIR'] if env['MLC_MLPERF_BACKEND'] == "ncnn":