Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add EfficientNet with pretrained weights #106

Closed
CarloLucibello opened this issue Feb 3, 2022 · 3 comments · Fixed by #171
Closed

Add EfficientNet with pretrained weights #106

CarloLucibello opened this issue Feb 3, 2022 · 3 comments · Fixed by #171

Comments

@CarloLucibello
Copy link
Member

@pxl-th has implemented EfficientNet models in his repo https://github.com/pxl-th/EfficientNet.jl and also implemented the infrastructure to load the weights from pytorch's checkpoints. If the author agrees, we could copy or move the code here and possibly extend the weight loading to other models.

@pxl-th
Copy link
Member

pxl-th commented Feb 3, 2022

Absolutely! I've wanted to contribute it for some time myself, but haven't managed yet :)

@CarloLucibello
Copy link
Member Author

thanks! Feel free to file a PR whenever you have time

@darsnack
Copy link
Member

darsnack commented Feb 7, 2022

I've added this to the central issue for tracking implementations/weights, so I will close this issue.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging a pull request may close this issue.

3 participants