Saving Pytorch model error: codec can't decode bytes in position 2-3

Question

我看到过关于加载 pytorch 模型的类似问题，但没有看到一个，而且针对这些问题提供的解决方案也没有解决我的问题。

这是我必须保存模型的代码：

PATH = "c:\Users\my_name\Desktop\model"
torch.save(model, PATH)

但我一直遇到标题错误。在评估之前，我在没有检查点的情况下保存模型。不确定这里出了什么问题。

Answer 1

最好只保存 state_dict。

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

并加载检查点：

model = TheModelClass(*args, **kwargs)
model.load_state_dict(torch.load(PATH))
model.eval()

Answer 2

如果路径只是我的 c: 驱动器，无论出于何种原因，错误都已修复。仍然不确定为什么我不能 select 更深入的路径。

deep-learning