PyTorch 错误加载保存的 nn.Module：对象没有属性“to”

Question

我使用的是 PyTorch 0.4。

我通过继承 nn.Module 定义了一个 PyTorch MyModel，并通过调用保存了它的一个实例

torch.save(my_model, my_path)

然后，当再次使用torch.load(my_path) 加载它时，我的程序崩溃并出现以下错误：

AttributeError: 'MyModel' object has no attribute 'to'

但我的程序能够在之前的阶段运行它。出了什么问题？

Answer 1

我已经找到了，只是想快速发布一下，因为谷歌没有给出明显的线索。

事实证明，虽然我从一台 0.4 的计算机上保存了模型，但我试图从另一台仍然安装了旧 (的计算机上加载它。 pip install --upgrade torch 已修复。

---

我发现它是因为my_model.train()和.eval()方法确实有效，所以我记得.to()方法是在0.4中引入的。有用的参考资料：

• https://pytorch.org/2018/04/22/0_4_0-migration-guide.html
• https://discuss.pytorch.org/t/loading-pytorch-model-without-a-code/12469