我正在训练一个pytorch模型,保存方式如下:
PATH = f"MODEL_trained.pt"
torch.save(model.state_dict(), PATH)
然后当我想加载它并执行它时,我使用这个:
model = torch.load(PATH)
但是,我得到了这个错误:
model.eval()
AttributeError: 'collections.OrderedDict' object has no attribute 'eval'
slackoverflow 上有很多这类问题,但是都没有解决我的问题。你能帮我吗?