Location via proxy:   [ UP ]  
[Report a bug]   [Manage cookies]                
Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

TorchVision: Add models with pre-trained weights. #588

Open
NiklasGustafsson opened this issue Apr 29, 2022 · 2 comments
Open

TorchVision: Add models with pre-trained weights. #588

NiklasGustafsson opened this issue Apr 29, 2022 · 2 comments

Comments

@NiklasGustafsson
Copy link
Contributor

We should add the following models from TorchVision, with the same pre-loaded weights:

https://pytorch.org/vision/stable/models.html#

See #586 for the origin story.

@NiklasGustafsson
Copy link
Contributor Author

Tracking: added torchvision models for ResNet, AlexNet, and VGG, all of which allow for "manual" loading of pre-trained weights. Instructions for doing so are in the doc comments.

@yueyinqiu
Copy link
Contributor

#713 (comment)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants