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

Implementing and training SE-ResNet and SE-ResNeXt and including them in "torchvision.models" #2179

Open
talcs opened this issue May 4, 2020 · 3 comments

Comments

@talcs
Copy link
Contributor

talcs commented May 4, 2020

🚀 Feature

Inclusion of pretrained SE-ResNet and SE-ResNeXt models in torchvision.

Motivation

SE-ResNet and SE-ResNeXt outperform ResNet and ResNeXt respectively and have been released around 2 years ago already.

Pitch

Implementing SE blocks and creating ResNet and ResNeXt models that include them, which will be called SE-ResNet and SE-ResNeXt and will be available as a part of torchvision.models. The SE paper has been published in CVPR 2018.

Alternatives

Additional context

@zhangguanheng66
Copy link
Contributor

@fmassa how do you think about those pretrained models?

@carlostojal
Copy link

Is this something still wanted? I am doing it either way, and if maintainers see this as a relevant addition I may open the pull request.

@NicolasHug
Copy link
Member

Thanks for offering to help @carlostojal - at this point we have no plan to include these models in torchvision

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

No branches or pull requests

4 participants