下载
以resnet50为例，设置pretrained=True就会下载权重
import torchvision.models as models
resnet50 = models.resnet50(pretrained=True)
默认存储路径：
C:\Users\Administrator\.torch\models
也可通过访问网址去下载：
'resnet18': 'https://download.pytorch.org/models/resnet18-5c106cde.pth',
'resnet34': 'https://download.pytorch.org/models/resnet34-333f7ec4.pth',
'resnet50': 'https://download.pytorch.org/models/resnet50-19c8e357.pth',
'resnet101': 'https://download.pytorch.org/models/resnet101-5d3b4d8f.pth',
'resnet152': 'https://download.pytorch.org/models/resnet152-b121ed2d.pth',
参考文献
官方文档：https://pytorch.org/docs/master/torchvision/models.html
官方代码：https://github.com/pytorch/vision/tree/master/torchvision/models
网友评论