search

pytorch / vision

Datasets, Transforms and Models specific to Computer Vision
https://pytorch.org/vision
BSD 3-Clause "New" or "Revised" License
15.99k stars 6.92k forks source link

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

Open talcs opened 4 years ago

talcs commented 4 years ago

🚀 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 commented 4 years ago

@fmassa how do you think about those pretrained models?

carlostojal commented 8 months ago

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 commented 8 months ago

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