Pytorch 加载预训练模型的两种方式：torchvision和torch.hub

从TorchVision中加载预训练模型

查看TorchVision中所包含的预训练模型

from torchvision import models

print(dir(models))

加载resnet18网络

from torchvision import models

resnet18 = models.resnet18(pretrained=True)

---

从Hub中加载预训练模型

查看Hub中所包含的预训练模型

model_list = torch.hub.list('pytorch/vision')

print(model_list)

加载resnet101网络

import torch

model = torch.hub.load('pytorch/vision', 'deeplabv3_resnet101', pretrianed=True)

---

两者区别

结合我的经历，最大的区别是，torchvision.models 中找不到的网络，可在torch.hub中找到

比如说senet, mobilenet_v1

但个人觉得torchvision.models 方便一点

---

推荐一个有关预训练模型加载不错的博客

pytorch最全预训练模型下载与调用