hub
hub copied to clipboard
Contribute a model to pytorh_hub
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
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 https://github.com/pytorch/hub/pull/219 which adds a bunch of models