question-mark
Stuck on an issue?

Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.

And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.

[Master Issue] Add more models to torchvision

See original GitHub issue

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.

Issue Analytics

  • State:open
  • Created 5 years ago
  • Reactions:8
  • Comments:46 (27 by maintainers)

github_iconTop GitHub Comments

3reactions
JuanFMontesinoscommented, Dec 13, 2018

Sooo let me evaluate it after christmas to see which dataset would be better

3reactions
guanfuchencommented, Dec 11, 2018

@fmassa VOC and Cityscapes dataset is large, there is a smaller dataset CamVid consisting 701 labeled images, and if you use U-Net for better performance, I think using the original medical dataset is better. Here is a good project named pytorch-semseg for semantic segmentation reimplementing U-Net.

Read more comments on GitHub >

github_iconTop Results From Across the Web

torchvision.models - PyTorch
The models subpackage contains definitions of models for addressing different tasks, including: image classification, pixelwise semantic segmentation, object ...
Read more >
Getting Started with PyTorch Image Models (timm)
The purpose of this guide is to explore timm from a practitioner's point of view, focusing on how to use some of the...
Read more >
Getting started with PyTorch - IBM
pytorch - Installs the GPU-enabled variants of PyTorch, torchvision, ... One or more elements of a deep learning model can lead to GPU...
Read more >
timm - PyPI
(Unofficial) PyTorch Image Models. ... Add new RelPosMlp MaxViT weight that leverages this: ... Two more model weights added in the TPU trained...
Read more >
torch.onnx — PyTorch master documentation
Using dictionaries to handle Named Arguments as model inputs. Indexing. Getter. Setter. TorchVision support. Limitations. Supported operators. Adding ...
Read more >

github_iconTop Related Medium Post

No results found

github_iconTop Related StackOverflow Question

No results found

github_iconTroubleshoot Live Code

Lightrun enables developers to add logs, metrics and snapshots to live code - no restarts or redeploys required.
Start Free

github_iconTop Related Reddit Thread

No results found

github_iconTop Related Hackernoon Post

No results found

github_iconTop Related Tweet

No results found

github_iconTop Related Dev.to Post

No results found

github_iconTop Related Hashnode Post

No results found