frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Run fp16 resnet50 training in bench script (#17831)

Browse Source 
Summary:
cc xw285cornell
Pull Request resolved: https://github.com/pytorch/pytorch/pull/17831

Differential Revision: D14398532

Pulled By: bddppq

fbshipit-source-id: 37c03cc2eebe3a6083e05631cb6ff03474e4a8a2
This commit is contained in:
Junjie Bai
2019-03-08 21:50:20 -08:00
committed by Facebook Github Bot
parent c10c73f047
commit 17232fb842
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
.jenkins/caffe2/bench.sh
View File

@ -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