模型下载
从pytorch官网找到对应模型的下载地址, 并使用浏览器下载 https://pytorch.org/hub/
模型加载(VGG11为例)
-
vgg11下载地址
https://github.com/pytorch/vision/blob/master/torchvision/models/vgg.py
-
本地加载
import torch import torchvision.models as models model = models.vgg11(pretrained=False) model.load_state_dict(torch.load(R'pytorch\vgg11-8a719046.pth')) print(model.eval()) # 输出 VGG( (features): Sequential( (0): Conv2d(3, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1)) (1): ReLU(inplace=True) (2): MaxPool2d(kernel_size=2, stride=2, padding=0, dilation=1, ceil_mode=False) (3): Conv2d(64, 128, kerne