Skip to content

Commit 7ae0e99

Browse files
authored
Fix model path for docker run (#868)
1 parent 518578b commit 7ae0e99

File tree

3 files changed

+7
-5
lines changed

3 files changed

+7
-5
lines changed

script/app-mlperf-inference/meta.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1720,7 +1720,7 @@ variations:
17201720
- "yes"
17211721
mounts:
17221722
- "${{ MLC_ML_MODEL_YOLOV11_PATH }}:${{ MLC_ML_MODEL_YOLOV11_PATH }}"
1723-
- "${{ MLC_DATASET_YOLO_COCO2017_FILTERED_PATH }}:${{ MLC_DATASET_YOLO_COCO2017_FILTERED_PATH }}"
1723+
- "${{ MLC_ML_DATASET_MLPERF_INFERENCE_YOLO_COCO2017_FILTERED_DATASET_PATH }}:${{ MLC_ML_DATASET_MLPERF_INFERENCE_YOLO_COCO2017_FILTERED_DATASET_PATH }}"
17241724
posthook_deps:
17251725
- enable_if_env:
17261726
MLC_MLPERF_LOADGEN_MODE:

script/get-dataset-mlperf-inference-yolo-coco2017-filtered-dataset/customize.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,8 +24,9 @@ def postprocess(i):
2424
env = i['env']
2525

2626
if env.get('MLC_DOWNLOAD_MODE', '') != 'dry':
27-
env['MLC_ML_DATASET_MLPERF_INFERENCE_YOLO_COCO2017_FILTERED_DATASET_PATH'] = os.path.join(
28-
env['MLC_ML_DATASET_MLPERF_INFERENCE_YOLO_COCO2017_FILTERED_DATASET_PATH'], "val2017_safe")
27+
if env.get('MLC_TMP_REQUIRE_DOWNLOAD', '') == "yes":
28+
env['MLC_ML_DATASET_MLPERF_INFERENCE_YOLO_COCO2017_FILTERED_DATASET_PATH'] = os.path.join(
29+
env['MLC_ML_DATASET_MLPERF_INFERENCE_YOLO_COCO2017_FILTERED_DATASET_PATH'], "val2017_safe")
2930
env['MLC_ML_DATASET_FILE_WITH_PATH'] = env['MLC_ML_DATASET_MLPERF_INFERENCE_YOLO_COCO2017_FILTERED_DATASET_PATH']
3031
env['MLC_ML_DATASET_MLPERF_INFERENCE_YOLO_COCO2017_FILTERED_DATASET_ANNOTATION_PATH'] = os.path.join(
3132
env['MLC_ML_DATASET_FILE_WITH_PATH'], "annotations", "instances_val2017.json")

script/get-ml-model-yolov11/customize.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,9 @@ def postprocess(i):
2323
env = i['env']
2424

2525
if env.get('MLC_DOWNLOAD_MODE', '') != 'dry':
26-
env['MLC_ML_MODEL_YOLOV11_PATH'] = os.path.join(
27-
env['MLC_ML_MODEL_YOLOV11_PATH'], 'yolo11l.pt')
26+
if env.get('MLC_TMP_REQUIRE_DOWNLOAD') == "yes":
27+
env['MLC_ML_MODEL_YOLOV11_PATH'] = os.path.join(
28+
env['MLC_ML_MODEL_YOLOV11_PATH'], 'yolo11l.pt')
2829
env['MLC_ML_MODEL_FILE_WITH_PATH'] = env['MLC_ML_MODEL_YOLOV11_PATH']
2930

3031
return {'return': 0}

0 commit comments

Comments
 (0)