File tree Expand file tree Collapse file tree 3 files changed +7
-5
lines changed
get-dataset-mlperf-inference-yolo-coco2017-filtered-dataset Expand file tree Collapse file tree 3 files changed +7
-5
lines changed Original file line number Diff line number Diff 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 :
Original file line number Diff line number Diff 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" )
Original file line number Diff line number Diff 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 }
You can’t perform that action at this time.
0 commit comments