pytorch / pytorch/vision

[Master Issue] Add more models to torchvision

Open
#645 46 comments 8 reactions 1 assignee View on GitHub

@fmassa is already working on this.

Since Oct 30, 2018.

enhancement
Dominant language
Python
Stars
17.9k
Forks
7.3k
Avg merge
1d 15h
Merged PRs (30d)
13

Description

This is a master issue to track requests for adding new pre-trained models to torchvision.

Here is the (potentially incomplete) list I compiled:

@Cadene has already implemented a number of those models in his fantastic https://github.com/Cadene/pretrained-models.pytorch . I'll start from there and try to get models trained using pytorch/examples/imagenet, so that the models are reproducible.


Requirements
  • python implementation to live in vision/models
  • pre-trained weights using the same mean / std normalization as in the imagenet example
  • the script used to train the models, or the command-line arguments used if the script was exactly the one from examples/imagenet.

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.