学习使用torch hub，但是遇到一些问题如： ImportError: cannot import name 'resnext50_32x4d'

之前知道torch hub这个库比较好用，而且看起来使用非常简单，但是自己一操作就报了标题错误;

torch版本我使用1.3.1, torchvison 却是0.2.1

顺着报错的提示找到torchvision.models 导入模型的时候缺少某些模型

于是选择更新torchvision 到最新版0.4.2

问题解决！

感觉torch hub 和torchvison.models 中导入模型有点像。

使用1：查看所有可用模型

import torch

model_list = torch.hub.list('pytorch/vision')
for model in model_list:
    print(model)

alexnet
deeplabv3_resnet101
densenet121
densenet161
densenet169
densenet201
fcn_resnet101
googlenet
inception_v3
mobilenet_v2
resnet101
resnet152
resnet18
resnet34
resnet50
shufflenet_v2_x0_5
shufflenet_v2_x1_0
squeezenet1_0
squeezenet1_1
vgg11
vgg11_bn
vgg13
vgg13_bn
vgg16
vgg16_bn
vgg19
vgg19_bn

 

使用2：模型导入，会直接下载下来模型

import torch

model = torch.hub.load('pytorch/vision', 'alexnet', pretrained=True)

model.eval()

使用3：一些方法

dir(model)

model.faward

model.state_dict

.......

待续。。。。