AttributeError：“函数”对象没有属性“复制”

Question

我正在尝试加载一个我在Google上训练过的模型state_dict，下面是加载模型的代码：

device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")

model = models.resnet50()
num_ftrs = model.fc.in_features
model.fc = nn.Linear(num_ftrs, n_classes)
model.load_state_dict(copy.deepcopy(torch.load("./models/model.pth",device)))
model = model.to(device)
model.eval()

以下是错误：

state_dict = state_dict.copy()

AttributeError：“函数”对象没有属性“复制”

毕圣火炬：

>>> import torch
>>> print (torch.__version__)
1.4.0
>>> import torchvision
>>> print (torchvision.__version__)
0.5.0

请帮忙，我到处找遍了，都没有用

全错误详细信息https://i.stack.imgur.com/s22DL.png

Answer 1

我猜这就是你错误地做了什么。你保存了函数

torch.save(model.state_dict, 'model_state.pth')

而不是state_dict()

torch.save(model.state_dict(), 'model_state.pth')

否则，一切都应如预期的那样运作。(我在Colab上测试了以下代码)

将model.state_dict()替换为model.state_dict以再现错误

import copy
model = TheModelClass()
torch.save(model.state_dict(), 'model_state.pth')
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
model.load_state_dict(copy.deepcopy(torch.load("model_state.pth",device)))

Answer 2

因为你救了你的模特

torch.save(model.state_dict, 'model_state.pth')

而不是

torch.save(model.state_dict(), 'model_state.pth')

因此，您保存了模型的函数指针。对于这个问题，您必须像这样加载数据：

model.load_state_dict(copy.deepcopy(torch.load("./models/model.pth",device)()))

而不是

model.load_state_dict(copy.deepcopy(torch.load("./models/model.pth",device)))

在torch.load("./models/model.pth",device)中，您可以看到模型层的详细信息，并获得其他良好的数据。