pytorch: 下载预训练模型

下载

以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