Reduced batch size to avoid out of memory condition in 19.07 containe…

…r. (#1089) Signed-off-by: Michal Zientkiewicz <[email protected]>
NVIDIA · Jul 19, 2019 · b9d05de · b9d05de
1 parent 401ea2a
commit b9d05de
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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