pytorch/resnet50_trainer.py at f93e0619f3e3e269e05aefd212b035da308404c3 - pytorch - Gitea: Git for Me

frozenleaves/pytorch

mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00

Files

Abhinav Jauhri f93e0619f3 Adding ShufflenetV2 to caffe2's benchmark suite. (#20180 )

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/20180

Adding ShufflenetV2 (by Ma et. al. 2018) to the caffe2's benchmark
suite.

To run, use: `buck run mode/opt caffe2/caffe2/python/examples:imagenet_trainer -- --train_data null --batch_size 128 --epoch_size 3200 --num_epochs 2 --num_gpus 2 --model shufflenet`

Reviewed By: bddppq, xw285cornell

Differential Revision: D15094282

fbshipit-source-id: 0e1ce9c5975868e917b0f179e2c5b15647a76b4e

2019-05-23 20:40:17 -07:00

Symbolic link

1 line

19 B

Python

Raw Blame History

imagenet_trainer.py