From 807632d40275586f2cddef8f7314b2011f288bfc Mon Sep 17 00:00:00 2001 From: Junjie Bai Date: Fri, 22 Feb 2019 15:01:46 -0800 Subject: [PATCH] Double resnet50 batch size in benchmark script (#17416) Summary: The benchmarks are now running on gpu cards with more memory Pull Request resolved: https://github.com/pytorch/pytorch/pull/17416 Differential Revision: D14190493 Pulled By: bddppq fbshipit-source-id: 66db1ca1fa693d24c24b9bc0185a6dd8a3337103 --- .jenkins/caffe2/bench.sh | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/.jenkins/caffe2/bench.sh b/.jenkins/caffe2/bench.sh index 1ba1cd7..6463cb5 100755 --- a/.jenkins/caffe2/bench.sh +++ b/.jenkins/caffe2/bench.sh @@ -17,17 +17,17 @@ fi caffe2_pypath="$(cd /usr && python -c 'import os; import caffe2; print(os.path.dirname(os.path.realpath(caffe2.__file__)))')" # Resnet50 if (( $num_gpus == 0 )); then - "$PYTHON" "$caffe2_pypath/python/examples/resnet50_trainer.py" --train_data null --batch_size 64 --epoch_size 6400 --num_epochs 2 --use_cpu + "$PYTHON" "$caffe2_pypath/python/examples/resnet50_trainer.py" --train_data null --batch_size 128 --epoch_size 12800 --num_epochs 2 --use_cpu fi if (( $num_gpus >= 1 )); then - "$PYTHON" "$caffe2_pypath/python/examples/resnet50_trainer.py" --train_data null --batch_size 64 --epoch_size 6400 --num_epochs 2 --num_gpus 1 + "$PYTHON" "$caffe2_pypath/python/examples/resnet50_trainer.py" --train_data null --batch_size 128 --epoch_size 12800 --num_epochs 2 --num_gpus 1 fi # Run multi-gpu training once the HSAQueue::isEmpty core dump issue is fixed # if (( $num_gpus >= 2 )); then -# "$PYTHON" "$caffe2_pypath/python/examples/resnet50_trainer.py" --train_data null --batch_size 128 --epoch_size 12800 --num_epochs 2 --num_gpus 2 +# "$PYTHON" "$caffe2_pypath/python/examples/resnet50_trainer.py" --train_data null --batch_size 256 --epoch_size 25600 --num_epochs 2 --num_gpus 2 # fi # if (( $num_gpus >= 4 )); then -# "$PYTHON" "$caffe2_pypath/python/examples/resnet50_trainer.py" --train_data null --batch_size 256 --epoch_size 25600 --num_epochs 2 --num_gpus 4 +# "$PYTHON" "$caffe2_pypath/python/examples/resnet50_trainer.py" --train_data null --batch_size 512 --epoch_size 51200 --num_epochs 2 --num_gpus 4 # fi # ResNext -- 2.7.4