From 17232fb8422e71da27a44c24721c7c0e352a91d8 Mon Sep 17 00:00:00 2001 From: Junjie Bai Date: Fri, 8 Mar 2019 21:50:20 -0800 Subject: [PATCH] Run fp16 resnet50 training in bench script (#17831) Summary: cc xw285cornell Pull Request resolved: https://github.com/pytorch/pytorch/pull/17831 Differential Revision: D14398532 Pulled By: bddppq fbshipit-source-id: 37c03cc2eebe3a6083e05631cb6ff03474e4a8a2 --- .jenkins/caffe2/bench.sh | 1 + 1 file changed, 1 insertion(+) diff --git a/.jenkins/caffe2/bench.sh b/.jenkins/caffe2/bench.sh index 6463cb5..c2f012d 100755 --- a/.jenkins/caffe2/bench.sh +++ b/.jenkins/caffe2/bench.sh @@ -21,6 +21,7 @@ if (( $num_gpus == 0 )); then fi if (( $num_gpus >= 1 )); then "$PYTHON" "$caffe2_pypath/python/examples/resnet50_trainer.py" --train_data null --batch_size 128 --epoch_size 12800 --num_epochs 2 --num_gpus 1 + "$PYTHON" "$caffe2_pypath/python/examples/resnet50_trainer.py" --train_data null --batch_size 256 --epoch_size 25600 --num_epochs 2 --num_gpus 1 --float16_compute --dtype float16 fi # Run multi-gpu training once the HSAQueue::isEmpty core dump issue is fixed # if (( $num_gpus >= 2 )); then -- 2.7.4