hub icon indicating copy to clipboard operation
hub copied to clipboard

Contribute a model to pytorh_hub

Open Riser6 opened this issue 3 years ago • 1 comments

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

Riser6 avatar Sep 02 '21 06:09 Riser6

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

NicolasHug avatar Sep 02 '21 10:09 NicolasHug