This is a unofficial Shake-Shake regularization implement of pytorch with Python3.
And boost much better than proposed using less time.
For train a model, please just use train scripts
Shake Shake model normally need to train 1800 epochs, but this repo just train 220 epochs, so if you need better results you could train a lettle more.
Pretrained model/log(s) is here
| Model | Proposed | This repo | Model size | Improved |
|---|---|---|---|---|
| epochs | 1800 | 220 | - | (save time) 818% |
| SSI 26 2x32d | 3.55 | 3.91 | 2.9M | -0.37% |
| SSI 26 2x64d | 2.98 | 2.90 | 11.7M | 0.08% |
| SSI 26 2x96d | 2.86 | 2.66 | 26.2M | 0.20% |
| SSI 26 2x112d | 2.82 | 2.56 | 35.6M | 0.26% |