代码之家 › 专栏 › 技术社区 › fr_andres

pytorch加载保存的nn时出错。模块:对象没有“to”属性

pytorch

fr_andres · 技术社区 · 8 年前

我用的是PyTorch 0.4 .

我定义了一个割炬 MyModel 继承自 nn.Module ,并通过调用

torch.save(my_model, my_path)

然后,当再次加载时 torch.load(my_path) ,我的程序因以下错误崩溃:

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

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

1 回复 | 直到 8 年前

fr_andres 8 年前

我已经发现了,只是想尽快发布,因为谷歌没有给出明显的线索。

结果是,虽然我从一台0.4的计算机上保存了模型,但我试图从另一台 仍然安装了较旧的(<0.4)pytorch版本 . pip install --upgrade torch 修好了。

我发现是因为 my_model.train() 和 .eval() 方法确实有效,所以我记得 .to() 方法在0.4中介绍。有用的参考资料:

推荐文章

Saffy · 如何在IterableDataset上应用最小最大缩放?

1 年前

sanjeev mk · 通过索引从Pytorch或Numpy 2D数组中快速删除多行的方法

2 年前

Anonymous · 如何为零维火炬张量赋值?

2 年前

JohnnyWang97 · getattr引起的有趣错误

2 年前

Kamugg · 在PyTorch中使用不同分辨率图像训练DeepLabV3的最佳实践

2 年前

Stocavista · 无法在python中将float 64转换为float 32

2 年前

efwefwefwefwefw wefwefwefwef · 如何在PyTorch Conv1d层中仅在一侧应用填充?

2 年前

Okhr · 运行时错误:CUDA错误:在带有GTX 1660 Super的Debian 12虚拟机上不支持此操作

2 年前

Fatemeh · 如何从使用nn训练和保存的模型加载检查点。DataParallel到不使用nn的模型上。DataParallel?

2 年前

Twenkid · 将GPT2 h5型号转换为割炬,以转换为ggml形状不匹配

2 年前