update mxnet dataloader threads

2021-10-03 01:53:09 -04:00 · 2021-10-03 01:53:09 -04:00 · 74b0f91f75
parent 3c964865ce
commit 74b0f91f75
1 changed files with 1 additions and 0 deletions
--- a/MxNet/Classification/RN50v1.5/MxNet_Imagenet.sh
+++ b/MxNet/Classification/RN50v1.5/MxNet_Imagenet.sh
@ -5,6 +5,7 @@ for (( i = 0; i < 3; i++ )); do # 3 runs
    ./runner -n 2 -b 192 --num-epochs 90 --mode train_val \
    --amp --dllogger-log /imagenet/MxNet/run_"${i}"/run_"${i}".log \
    --workspace /imagenet/MxNet/run_"${i}"/ --data-backend mxnet \
+    --data-backend-threads 80 \
    --data-root /imagenet --warmup 0
    echo "Run ${i} done"
 done