search

dotnet / TorchSharp

A .NET library that provides access to the library that powers PyTorch.
MIT License
1.37k stars 177 forks source link

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

Open NiklasGustafsson opened 2 years ago

NiklasGustafsson commented 2 years ago

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 commented 2 years ago

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 commented 5 months ago

https://github.com/dotnet/TorchSharp/issues/713#issuecomment-2064095832