Home

Awesome

PyTorch Hub

CircleCI

Logistics

We accept submission to PyTorch hub through PR in hub repo. Once the PR is merged into master here, it will show up on the PyTorch website in 24 hrs.

Steps to submit to PyTorch hub

  1. Add a hubconf.py in your repo, following the instruction in torch.hub doc. Verify it's working correctly by running torch.hub.load(...) locally.
  2. Create a PR in pytorch/hub repo. For each new model you have, create a <repo_owner>_<repo_name>_<title>.md file using this template.

Notes

python scripts/sanity_check.py
./scripts/run_pytorch.sh