TypeError:使用模型时无法调用“ NoneType”对象

时间:2020-11-08 23:08:44

标签: model pytorch call nonetype

我正在使用PyTorch训练GCN,我有一个使用以下命令保存的简单模型:

torch.save(MyNet().state_dict(), PATH)

然后我通过执行以下操作加载它:

model = MyNet()
model.load_state_dict(torch.load(PATH))
model.eval()

但是当我尝试如下输入数据时:

output = model(dataset)

上一行出现此错误:

TypeError: 'NoneType' object is not callable

我不知道是什么原因,请您能帮我吗?

0 个答案:

没有答案