我保存并加载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
,然后对其进行重新培训而不会出现错误?