手动修改pytorch state_dict文件并重新训练

时间:2019-04-29 09:57:15

标签: pytorch

我保存并加载pytorch state_dict文件,然后评估并重新训练模型。这样可行。

torch.save(net.state_dict(), path)
net.load_state_dict(torch.load(path))

但是,当我在加载并评估state_dict文件后(手动更改值)进行修改时,会收到错误消息:

RuntimeError: a leaf Variable that requires grad has been used in an in-place operation.

如何安全地修改state_dict,然后对其进行重新培训而不会出现错误?

0 个答案:

没有答案