AojunZhou / pytorch_ShuffleNet

pytorch version of ShuffleNet

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

ShuffleNet with PyTorch

Note: This project is pytorch implementation of ShuffleNet.

Performance

Trained on ImageNet with groups=3, get Prec@1 67.898% and Prec@5 87.994%. During the training, I set batch_size=256, learning_rate=0.1 which decayed every 30 epoch by 10.

Training on ImageNet

python main -b 256 $Imagenetdir

License: MIT license (MIT)

About

pytorch version of ShuffleNet


Languages

Language:Python 100.0%