Commit f93e061
Adding ShufflenetV2 to caffe2's benchmark suite. (#20180)
Summary:
Pull Request resolved: #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: 0e1ce9c5975868e917b0f179e2c5b15647a76b4e1 parent 3aa7ee6 commit f93e061
File tree
6 files changed
+1207
-856
lines changed- caffe2/python
- examples
- models
6 files changed
+1207
-856
lines changed
0 commit comments