File tree Expand file tree Collapse file tree 2 files changed +6
-2
lines changed
script/get-dataset-coco2014 Expand file tree Collapse file tree 2 files changed +6
-2
lines changed Original file line number Diff line number Diff line change @@ -9,6 +9,9 @@ cache: true
9
9
category : AI/ML datasets
10
10
category_sort : 8500
11
11
12
+ input_mapping :
13
+ num_workers : MLC_DATASET_COCO2014_NUM_WORKERS
14
+
12
15
tags :
13
16
- get
14
17
- dataset
18
21
19
22
default_env :
20
23
MLC_DATASET_CALIBRATION : ' no'
21
-
24
+ MLC_DATASET_COCO2014_NUM_WORKERS : 1
25
+
22
26
deps :
23
27
24
28
- names :
Original file line number Diff line number Diff line change @@ -27,7 +27,7 @@ if [[ ${MLC_DATASET_CALIBRATION} == "no" ]]; then
27
27
eval $cmd
28
28
test $? -eq 0 || exit $?
29
29
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} "
31
31
echo $cmd
32
32
eval $cmd
33
33
test $? -eq 0 || exit $?
You can’t perform that action at this time.
0 commit comments