TL;DR: Implemented EfficientNet in PyTorch: https://github.com/lukemelas/EfficientNet-PyTorch

​

I found EfficientNet to be quite an exciting paper, so I implemented it in PyTorch over the past two days. The implementation makes it simple to load pretrained models and integrate them into your own projects/research/models. GitHub: https://github.com/lukemelas/EfficientNet-PyTorch

​

There are multiple examples in the GitHub repo and here is one on Colab. It’s as quick as

from efficientnet_pytorch import EfficientNet model = EfficientNet.from_pretrained('efficientnet-b0') 

And you can install it via pip if you would like:

pip install efficientnet_pytorch 

Finally, there are scripts to evaluate on ImageNet (with training scripts coming soon) and there’s functionality to easily extract image features.

​

Let me know (either here or on GitHub) if you have any comments or find any bugs. I hope some of you find it useful!