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

Contribute a model to pytorh_hub #226

Open
Riser6 opened this issue Sep 2, 2021 · 1 comment
Open

Contribute a model to pytorh_hub #226

Riser6 opened this issue Sep 2, 2021 · 1 comment

Comments

@Riser6
Copy link
Contributor

Riser6 commented Sep 2, 2021

Hi, thanks for providing so many amazing AI model for us AI researchers!
Currently, I want to provide our model (https://github.com/hustvl/YOLOP) to PyTorch Hub. And I have added the hubconf.py to my repository and filled the project information in hub/docs/template.md, But there seems to be no news. Did I miss anything? Appreciate very much if you can answer it!

Regards

@NicolasHug
Copy link
Member

Hi @Riser6 , thank you for your interest in publishing your model on torchhub.

I'm looking at your hubconf.py, and I think the __main__ function etc. are unnecessary. Take a look for example at torchvision's hubconf.py: all you need to do is to expose callables / objects in this file. You can read more about this in our docs: https://pytorch.org/docs/stable/hub.html

and filled the project information in hub/docs/template.md

For you model to show up on the website you will need to submit a PR with this file in this repo: https://github.com/pytorch/hub. See for example this PR #219 which adds a bunch of models

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

No branches or pull requests

2 participants