search

Cadene / pretrained-models.pytorch

Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.
BSD 3-Clause "New" or "Revised" License
9.01k stars 1.83k forks source link

Shake-shake regularization Net #32

Open ahkarami opened 6 years ago

ahkarami commented 6 years ago

Dear @Cadene, Do you have any plan to re-implement or convert the Shake-shake regularization Net into PyTorch and your nice repository? To the best of my knowledge, it has the best performance on cifar 10 currently.

Cadene commented 6 years ago

Hi @ahkarami,

Having pretrained models on different datasets than imagenet would be great. However, I don't see any pretrained models on this repo https://github.com/xgastaldi/shake-shake.

If you already have one, you should try to convert it to pytorch using this awesome tool: https://github.com/clcarwin/convert_torch_to_pytorch