pytorch/caffe2/python/models
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
..
seq2seq Unify cuda and hip device types in Caffe2 python front end (#14221) 2018-11-29 14:00:16 -08:00
__init__.py Experimental support for setup.py develop mode install 2018-02-12 23:36:18 -08:00
__sym_init__.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
download.py Add debug logic to c2_ref_test and its helpers (#19359) 2019-04-22 12:08:55 -07:00
imagenet_trainer_test_utils.py Adding ShufflenetV2 to caffe2's benchmark suite. (#20180) 2019-05-23 20:40:17 -07:00
resnet_test.py Adding ShufflenetV2 to caffe2's benchmark suite. (#20180) 2019-05-23 20:40:17 -07:00
resnet.py Add resnext model to OSS (#11468) 2018-09-12 15:59:20 -07:00
shufflenet_test.py Adding ShufflenetV2 to caffe2's benchmark suite. (#20180) 2019-05-23 20:40:17 -07:00
shufflenet.py Adding ShufflenetV2 to caffe2's benchmark suite. (#20180) 2019-05-23 20:40:17 -07:00