我正在使用PyTorch 0.4 。
我通过继承MyModel定义了一个PyTorch nn.Module，并通过调用
torch.save(my_model, my_path)
然后，当用torch.load(my_path)再次加载它时，我的程序崩溃并出现以下错误：
AttributeError: 'MyModel' object has no attribute 'to'
但是我的程序能够在之前的阶段中运行它。出了什么问题？
答案 0 :(得分：3)
我已经找到了它，并且只是想快速发布有关它的信息，因为google没有给出明显的线索。
事实证明，尽管我从装有0.4的计算机上保存了模型，但是我试图从另一台仍安装了旧版本（<0.4）PyTorch的计算机上加载模型。 pip install --upgrade torch对其进行了修复。
我发现是因为my_model.train()和.eval()方法确实有效，所以我记得.to()方法是在0.4中引入的。有用的参考文献：