相关疑难解决方法(0)

我有同样的问题我遇到了与How can I load and use a PyTorch (.pth.tar) model该模型没有公认的答案，或者我可以弄清楚如何遵循给出的建议。

我是 PyTorch 的新手。我正在尝试加载此处引用的预训练 PyTorch 模型： https: //github.com/macaodha/inat_comp_2018

我很确定我缺少一些胶水。

# load the model
import torch
model=torch.load("iNat_2018_InceptionV3.pth.tar",map_location='cpu')

# try to get it to classify an image
imsize = 256
loader = transforms.Compose([transforms.Scale(imsize), transforms.ToTensor()])

def image_loader(image_name):
    """load image, returns cuda tensor"""
    image = Image.open(image_name)
    image = loader(image).float()
    image = Variable(image, requires_grad=True)
    image = image.unsqueeze(0)  
    return image.cpu()  #assumes that you're using CPU

image = image_loader("test-image.jpg")

产生错误：

in () ----> 1 model.predict(image)

AttributeError：“dict”对象没有属性“predict”