kuangliu / pytorch-cifar

95.47% on CIFAR10 with PyTorch
MIT License
5.94k stars 2.14k forks source link

Suggestion for MobileNetV2 #133

Open nlp-tokio opened 3 years ago

nlp-tokio commented 3 years ago

I found that the MobileNetv2 has more shortcut with conv1d and batchnorm ops when stride==1 compared to original paper, which leads to the high performance of the modified model. I think extra comment could be added into MobileNetv2 code file for other guys.