我正在使用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
我不知道是什么原因,请您能帮我吗?