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

Models not pushed to torch hub #4289

Open
2 of 6 tasks
oke-aditya opened this issue Aug 18, 2021 · 4 comments
Open
2 of 6 tasks

Models not pushed to torch hub #4289

oke-aditya opened this issue Aug 18, 2021 · 4 comments

Comments

@oke-aditya
Copy link
Contributor

oke-aditya commented Aug 18, 2021

📚 Documentation

Hi!

The following models are listed in our hubconf.py
but these models are not listed over pytorch hub

  • MobileNetv3: - Mobilenet v3 small, Mobilenet v3 large
  • Mobilenet Segmentation: - lraspp_mobilenet_v3_large
  • DeepLab Models: - deeplabv3_mobilenet_v3_large deeplabv3_resnet50
  • FCN: - fcn_resnet50,
  • MNasNet: - mnasnet0_5, mnasnet0_75, mnasnet1_0, mnasnet1_3
  • EfficientNet: - efficientnet_b0, efficientnet_b1, efficientnet_b2, efficientnet_b3, efficientnet_b4, efficientnet_b5, efficientnet_b6, efficientnet_b7

Additional Context: -

I think there is ongoing issue with torch hub. #4156
Also we do not publish detection models to torch hub see #1945

P.S.
Please verify the list and feel to free to edit if I have missed something.
Also probably this issue can be transferred to torch hub repository.

cc @datumbox @NicolasHug

@datumbox
Copy link
Contributor

Thanks for bringing this up @oke-aditya. I think it's reasonable to include them.

The only concern I got is for mnasnet0_75 and mnasnet1_3 which don't have pre-trained weights, but all other look good to me. Would you like to send a PR?

@oke-aditya
Copy link
Contributor Author

Yes, I will! Happy to help 😄

@NicolasHug
Copy link
Member

Thanks @oke-aditya ,

as a minor caveat we should just make sure that those models aren't relying on custom ops, which is the reason why the detection models aren't published yet: #1945 (comment)

(I haven't checked myself)

@datumbox
Copy link
Contributor

All models listed here don't require require custom operators that need compilation. He should be good to go.

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

3 participants