mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
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 |
||
|---|---|---|
| .. | ||
| seq2seq | ||
| __init__.py | ||
| __sym_init__.py | ||
| download.py | ||
| imagenet_trainer_test_utils.py | ||
| resnet_test.py | ||
| resnet.py | ||
| shufflenet_test.py | ||
| shufflenet.py | ||