diff --git a/qa/TL3_RN50_convergence/test_mxnet.sh b/qa/TL3_RN50_convergence/test_mxnet.sh index bda57ae305..ca4819671c 100644 --- a/qa/TL3_RN50_convergence/test_mxnet.sh +++ b/qa/TL3_RN50_convergence/test_mxnet.sh @@ -6,7 +6,7 @@ min_perf=10000 NUM_GPUS=`nvidia-smi -L | wc -l` -python /opt/mxnet/example/image-classification/train_imagenet_runner -n $NUM_GPUS --seed 42 2>&1 | tee dali.log +python /opt/mxnet/example/image-classification/train_imagenet_runner -b 208 -n $NUM_GPUS --seed 42 2>&1 | tee dali.log cat dali.log | grep -o "Validation-accuracy=0\.[0-9]*" > tmp2.log cat dali.log | grep -o "Speed: [0-9]*\.[0-9]*" > tmp3.log