Skip to content

Commit b808608

Browse files
authored
Minor fixes to improve submission generation experience (#242)
1 parent dd6f2a8 commit b808608

File tree

3 files changed

+3
-2
lines changed

3 files changed

+3
-2
lines changed

script/app-mlperf-inference-mlcommons-python/meta.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -885,6 +885,7 @@ variations:
885885
ENQUEUE_NUM_THREADS: 2
886886
MLC_MLPERF_LOADGEN_MAX_BATCHSIZE: 16
887887
MLC_MLPERF_VISION_DATASET_OPTION: imagenet_pytorch
888+
MLC_ML_MODEL_STARTING_WEIGHTS_FILENAME: <<<MLC_MLPERF_CUSTOM_MODEL_PATH>>>
888889

889890

890891
tvm-onnx:

script/app-mlperf-inference/customize.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -254,7 +254,7 @@ def postprocess(i):
254254
measurements['starting_weights_filename'] = env.get(
255255
'MLC_ML_MODEL_STARTING_WEIGHTS_FILENAME', env.get(
256256
'MLC_ML_MODEL_FILE', measurements.get(
257-
'starting_weights_filename', '')))
257+
'starting_weights_filename', 'TBD')))
258258
measurements['retraining'] = env.get(
259259
'MLC_ML_MODEL_RETRAINING', measurements.get(
260260
'retraining', 'no'))

script/app-mlperf-inference/meta.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ default_env:
2727
MLC_MLPERF_RUN_STYLE: test
2828
MLC_TEST_QUERY_COUNT: '10'
2929
MLC_MLPERF_QUANTIZATION: off
30-
MLC_GET_PLATFORM_DETAILS: yes
30+
MLC_GET_PLATFORM_DETAILS: no
3131

3232
env:
3333
MLC_MLPERF_PRINT_SUMMARY: "no"

0 commit comments

Comments
 (0)