diff --git a/script/get-dataset-coco2014/meta.yaml b/script/get-dataset-coco2014/meta.yaml index fa3724f83..afdff9cc4 100644 --- a/script/get-dataset-coco2014/meta.yaml +++ b/script/get-dataset-coco2014/meta.yaml @@ -9,6 +9,9 @@ cache: true category: AI/ML datasets category_sort: 8500 +input_mapping: + num_workers: MLC_DATASET_COCO2014_NUM_WORKERS + tags: - get - dataset @@ -18,7 +21,8 @@ tags: default_env: MLC_DATASET_CALIBRATION: 'no' - + MLC_DATASET_COCO2014_NUM_WORKERS: 1 + deps: - names: diff --git a/script/get-dataset-coco2014/run.sh b/script/get-dataset-coco2014/run.sh index a891b2330..21829568e 100644 --- a/script/get-dataset-coco2014/run.sh +++ b/script/get-dataset-coco2014/run.sh @@ -27,7 +27,7 @@ if [[ ${MLC_DATASET_CALIBRATION} == "no" ]]; then eval $cmd test $? -eq 0 || exit $? else - cmd="./download-coco-2014-calibration.sh -d ${INSTALL_DIR}" + cmd="./download-coco-2014-calibration.sh -d ${INSTALL_DIR} -n ${MLC_DATASET_COCO2014_NUM_WORKERS}" echo $cmd eval $cmd test $? -eq 0 || exit $?