Pytorch中torch.load()中出现AttributeError: Can‘t get attribute

本文介绍了解决PyTorch中使用torch.load()时遇到的AttributeError问题,该问题源于Pickle处理保存/加载模型时无法找到类定义。文章提供了具体的解决方法,即确保类定义可见。

低功耗蓝牙项目,需要一块懂省电的板

思澈 SF32LB52 芯片,BLE 协议栈深度优化,上手即开发

Pytorch中torch.load()中出现AttributeError: Can't get attribute

原因:保存下来的模型和参数不能在没有类定义时直接使用。

Pytorch使用Pickle来处理保存/加载模型,这个问题实际上是Pickle的问题,而不是Pytorch。

解决方法也非常简单,只需显式地导入类定义。即将包含类定义的文件复制粘贴到与要运行的文件同一文件夹下,再import Class!

但是,在实际过程中,我们采取import Class的方法没起到作用,而直接在要运行的文件上复制类定义是一种可行的方法。

低功耗蓝牙项目,需要一块懂省电的板

思澈 SF32LB52 芯片,BLE 协议栈深度优化,上手即开发

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值