Skip to content
Permalink
master
Switch branches/tags
Go to file
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: 0e1ce9c5975868e917b0f179e2c5b15647a76b4e
12 contributors

Users who have contributed to this file

@akyrola @pietern @lukeyeager @bddppq @harouwu @rohithkrn @orionr @Yangqing @tomdz @manofmountain @wesolwsk @HengCV
imagenet_trainer.py