加载模型问题;Pytorch

Question

我正在尝试加载一个保存了以下内容的模型，它给了我这个错误：

import torch
model = torch.load('./grunet.pkl')
model.eval()

这就是错误：

Traceback (most recent call last):
File ".eval.py", line 4, in <module>
model.eval()
AttributeError: 'NoneType' object has no attribute 'eval'

请帮忙！

Answer 1

您缺少一个步骤：

model = YourModelClass()
model.load_state_dict(torch.load("./grunet.pkl"))
model.eval()
# do something with the model