Skip to content

Commit cfa5519

Browse files
authored
Fix argument issue in coco2014 calibration dataset download (#286)
* add num of workers * Update run.sh
1 parent 7a5f1e1 commit cfa5519

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

script/get-dataset-coco2014/meta.yaml

+5-1
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,9 @@ cache: true
99
category: AI/ML datasets
1010
category_sort: 8500
1111

12+
input_mapping:
13+
num_workers: MLC_DATASET_COCO2014_NUM_WORKERS
14+
1215
tags:
1316
- get
1417
- dataset
@@ -18,7 +21,8 @@ tags:
1821

1922
default_env:
2023
MLC_DATASET_CALIBRATION: 'no'
21-
24+
MLC_DATASET_COCO2014_NUM_WORKERS: 1
25+
2226
deps:
2327

2428
- names:

script/get-dataset-coco2014/run.sh

+1-1
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ if [[ ${MLC_DATASET_CALIBRATION} == "no" ]]; then
2727
eval $cmd
2828
test $? -eq 0 || exit $?
2929
else
30-
cmd="./download-coco-2014-calibration.sh -d ${INSTALL_DIR}"
30+
cmd="./download-coco-2014-calibration.sh -d ${INSTALL_DIR} -n ${MLC_DATASET_COCO2014_NUM_WORKERS}"
3131
echo $cmd
3232
eval $cmd
3333
test $? -eq 0 || exit $?

0 commit comments

Comments
 (0)