Skip to content
No description, website, or topics provided.
Branch: master
Clone or download
ailzhang Merge pull request #38 from pytorch/debug_preview
Remove cached folder in preview
Latest commit 1998d5e Jun 14, 2019
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
.circleci Update config.yml Jun 12, 2019
docs Update template.md Jun 11, 2019
images added unet for brain mri segmentation (#33) Jun 13, 2019
scripts Update install.sh Jun 11, 2019
.ruby-version found minimal set of required deps file Jun 3, 2019
README.md fix CI & a few improvements Jun 6, 2019
dog.jpg add dog from wikipedia Jun 2, 2019
facebookresearch_pytorch-gan-zoo_dcgan.md add accelerator tag Jun 10, 2019
facebookresearch_pytorch-gan-zoo_pgan.md add accelerator tag Jun 10, 2019
huggingface_pytorch-pretrained-bert_bert.md add accelerator tag Jun 10, 2019
huggingface_pytorch-pretrained-bert_gpt.md add accelerator tag Jun 10, 2019
mateuszbuda_brain-segmentation-pytorch_unet.md added unet for brain mri segmentation (#33) Jun 13, 2019
nvidia_deeplearningexamples_tacotron2.md reorder code snippets Jun 12, 2019
nvidia_deeplearningexamples_waveglow.md reorder code snippets Jun 12, 2019
package.json bring back package.json; try removing .nvmrc Jun 3, 2019
preview_setup.sh remove old clone of pytorch.github.io Jun 14, 2019
pytorch_vision_alexnet.md add accelerator tag Jun 10, 2019
pytorch_vision_deeplabv3_resnet101.md fix typo (#32) Jun 11, 2019
pytorch_vision_densenet.md add accelerator tag Jun 10, 2019
pytorch_vision_fcn_resnet101.md add accelerator tag Jun 10, 2019
pytorch_vision_googlenet.md add accelerator tag Jun 10, 2019
pytorch_vision_inception_v3.md add accelerator tag Jun 10, 2019
pytorch_vision_mobilenet_v2.md add accelerator tag Jun 10, 2019
pytorch_vision_resnet.md add accelerator tag Jun 10, 2019
pytorch_vision_resnext.md add accelerator tag Jun 10, 2019
pytorch_vision_shufflenet_v2.md add accelerator tag Jun 10, 2019
pytorch_vision_squeezenet.md add accelerator tag Jun 10, 2019
pytorch_vision_vgg.md add accelerator tag Jun 10, 2019
yarn.lock bring back yarn.lock; try removing Gemfile Jun 3, 2019

README.md

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 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

  • Currently we don't support hosting pretrained weights, users with pretrained weights need to host them properly themselves.
  • In general we recommend one model per markdown file, models with similar structures like resnet18, resnet50 should be placed in the same file.
  • If you have images, place them in images/ folder and link them correctly in the [images/featured_image_1/featured_image_2] fields above.
  • We only support a pre-defined set of tags, currently they are listed in scripts/tags.py. We accept PRs to expand this set as needed.
  • To test your PR locally, run the tests below.
python scripts/sanity_check.py
./scripts/run_pytorch.sh
  • Our CI concatenates all python code blocks in one markdown file and runs it agaist the latest PyTorch release.
    • Remember to mark your python code using ```python in model markdown file.
    • If your dependencies is not installed on our CI machine, add them in install.sh.
    • If it fails, you can find a new temp.py file left in the repo to reproduce the failure.
  • We also provide a way to preview your model webpage through netlify bot. This bot builds your PR with latest pytorch.github.io repo and comment on your PR with preview link. The preview will be updated as you push more commits to the PR. Example netlify bot comment
You can’t perform that action at this time.