之前知道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
.......
待续。。。。