pytorch加载保存的模型报错

最新推荐文章于 2024-07-18 10:25:43 发布
最新推荐文章于 2024-07-18 10:25:43 发布
阅读量562 收藏
点赞数 1
文章标签： 深度学习
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/mzm123321/article/details/106764802
版权

使用pytorch加载保存的模型时报错：AttributeError: Can't get attribute 'CNN' on <module '__main__' 

torch.save(CNN, PATH)

CNN= torch.load(PATH)

解决：在同一个py文件内，将声明 CNN 类的代码加上即可

 

未央君@
关注
【pytorch】load模型报错
andeyeluguo的博客
08-08 81
有时候load模型报错，可能是pth文件不完整，因为wget下载文件即使下载文件没下载完也是那个文件名，所以都需要md5或者sha来判断是否完整，解决方法是重新下载。
pytorch加载模型报错Unexpected key(s) in state_dict: module.conv1.weight, module.bn1
热门推荐
Hello Word!
06-28 1万+
文章目录代码报错原因解决 代码 import models arch = 'resnet50' model = models.__dict__[arch]() checkpoint = torch.load(ckptFile) model.load_state_dict(checkpoint['state_dict']) model = torch.nn.DataParallel(model).cuda() 报错 Traceback (most recent call last
使用torch.load时，出现错误：Can‘t get attribute ‘Net‘ on module ‘__main__‘
最新发布
Genius_in_tears的博客
07-18 455
【代码】使用torch.load时，出现错误：Can‘t get attribute ‘Net‘ on module ‘__main__‘
Pytorch加载模型时报错
carlwuuu的博客
12-28 1779
加载模型时出现Unexpected key(s) in state_dict错误 报错截图如下： 反复排查问题没发现为何如此，查看pytorch中文文档发现保存和加载模型方法都完全正确，模型保存和加载代码对比中文文档截图如下： 其中一个方法是在加载模型时添加参数strict=False，可以只保留键值相同的参数避免出错，用法如下：model.load_state_dict(ckpt[‘stat...
torch 变量_[错误汇总]torch.load加载模型报错；cuda数据类型报错
weixin_39927214的博客
11-20 732
torch.load加载模型报错 ：UnicodeDecodeError: 'ascii' codec can't decode byte 0xc3 in position 1124: ordinal not in range(128)原因是：Pytorch在加载模型时候，如果训练保存模型的python环境是2.7，然而在python3的环境中去load模型，就会出现上面的错误。解决办法：asci...
torch.load AttributeError: Can‘t get attributet
追梦小狂魔的博客
02-18 1854
出现这个问题是因为直接使用了torch.load没有导入相应的模型文件 from mymodels import * 导入定义模型的py文件即可
解决pytouch导入模型torch.load报错：AttributeError: Can‘t get attribute ‘XXX‘ on ＜module ‘__main__‘ from XXX＞
levelina的博客
01-02 1376
解决办法：在torch.load（）之前，加入你定义的模型的CLASS，也就是把训练模型时的定义的模型的类复制到torch.load（）之前。
torch.load时报AttributeError: Can‘t get attribute ‘Module‘ on ＜module ‘__main__‘错误的解决方法
人工智能
05-11 3080
这里就是由于我们在test_1.py里定义了Module这个类，在序列化的时候也把这个类序列化进去了，当我们在另一个文件中加载的时候，由于没有Module这个类在内存里，就会导致找不到Module这个模型类进行反序列化成对象。torch.save()函数使用的是google的protobuf协议进行序列化的，当我们在save一个已经存在的Module的时候，在另一个py文件中load是没有问题的。的时候，问题也是一样，按照上面的方式解决就行！这样就可以正常运行了。
Pytorch加载保存好的模型发现与实际保存模型的参数不一致
weixin_42416791的博客
09-03 9423
Pytorch加载保存好的模型发现与实际保存的参数不一致发现问题初衷：问题：保存模型并打印参数：加载模型并打印模型参数问题来了解决问题先说结论原因：联想问题本质二级目录三级目录参考博客 发现问题 初衷： 最近跑深度学习代码，数据量比较庞大所以训练速度慢的一，一旦有什么问题训练突然停止了，心态都要炸了，所以这时候使用pytorch保存每轮训练好的模型（可以使用一些保存技巧，保存得到的当前最新的模型checkpoint，倒不至于每轮都保存），即使训练停止还能接着去加载保存好的模型得到当前模型输出结果，节省时间，
Pytorch保存的模型加载失败问题
weixin_42915923的博客
10-13 1115
训练好的模型在另外一台电脑上加载时出现如下错误： 两台电脑的Pytorch、Pytorch、Spyder版本都是一样的，只不过报错的这个电脑的CUDA是11.3，跟Pytorch官网上支持的11.1有区别，虽然我在安装后测试时是可以正常使用的，但这有可能是潜在的问题来源。 网上的解决方法是将模型的Class import一下，但是我这里是已经import过的。 所以我这里将原电脑…/modules/linear.py替换了报错电脑的相同文件，然后不在报错。 ...
pytorch预训练模型报错HTTPSConnectionPool(host='huggingface.co', port=443
10-08
这个错误通常是由于网络连接问题引起的。请确保你的网络连接正常，并检查是否可以访问 "huggingface.co" 的网站。...如果以上方法都无法解决问题，你可以考虑下载预训练模型的源文件，并手动加载到你的程序中。
AttributeError Can‘t get attribute ‘Model‘ on module ‘main‘
Akun_2217的博客
12-01 646
torch.load时会出现这种错误，本质是在当前文件内运行时并没有发现Model类。AttributeError经常是说明调用了某一个类没有的属性。
【Pytorch】使用torch.load()加载模型时报错，提示加载的模型是zip格式
qq_44846512的博客
07-12 2301
问题 在pytorch版本1.6.0上保存最优模型，当将该模型下载到其他设备上时发现模型从.pt转成了zip压缩包格式或者直接使用在其他版本的pytorch上加载时报错。 解决方法 torch官网上指出 The 1.6 release of PyTorch switched torch.save to use a new zipfile-based file format. torch.load still retains the ability to load files in the old forma
[pytorch] 模型加载，torch.load , keys not match
weixin_43918682的博客
03-26 1977
可能会遇到有一个多gpu的训练后保存的模型，但是后续要在单gpu的机子上使用，在torxh.load时报错。 这是因为nn.DataParallel会在模型参数结构前面加一个module. 比如你是这么save的 state = {'epoch': epoch, 'state_dict': self.model.state_dict(), 'optimizer': self.optimizer.state_dict(), '
torch.load加载模型报错：Can‘t get attribute ‘vae_vc‘ on ＜module ‘__main__‘ from ‘xxxx()运行文件路径‘
oneself的博客
06-29 1317
报错原因：torch.load在加载模型时实际需要参考网络结构构建出一个对应网络的实例对象，需要参考源class
Pytorch中torch.load()中出现AttributeError: Can‘t get attribute
wangchaoxjtu的专栏
05-17 4298
Pytorch中torch.load()中出现AttributeError: Can't get attribute 原因：保存下来的模型和参数不能在没有类定义时直接使用。 Pytorch使用Pickle来处理保存/加载模型，这个问题实际上是Pickle的问题，而不是Pytorch。 解决方法也非常简单，只需显式地导入类定义。即将包含类定义的文件复制粘贴到与要运行的文件同一文件夹下，再import Class！ 但是，在实际过程中，我们采取import Class的方法没起到作用，而直接在要运行的文
pytorch加载模型时报错
qq_39076073的博客
06-03 377
pyorch加载模型是出现错误： model.load_state_dict(checkpoint['state_dict']) 错误： RuntimeError: Error(s) in loading state_dict for SiameseNet: Missing key(s) in state_dict: "CNN.pretrained.conv1.weight", "CNN.pretrained.bn1.weight", "CNN.pretrained.bn1.bias", "CNN.pr
torch.load加载模型后出现问题
qq_39511822的博客
06-04 1444
torch.load后模型效果非常差
自学pytorch深度学习遇到的坑之模型加载出错
weixin_42233120的博客
09-09 2488
当保存了神经网络模型后，进行训练加载模型的时候一直出错，代码如下： model = model.load_state_dict(torch.load(path)) 运行结果如下，出现： Error(s) in loading state_dict for ResNet18 出现这样的问题，应该是版本的原因，这样的语句现在新的版本好像不支持了（如果有朋友知道具体的原因，欢迎留言指正哈……） 这...
pytorch：使用pytorch加载模型时报错
Jcsim~
01-28 1068
在utils.py里调用doCheakByBytes(image)，该方法里有torch模型的加载，报错AttributeError: Can't get attribute 'Model' on <module '__main__' from 'xxxxx/utils.py'> 解决方法：在utils.py文件里import Model 感谢：https://blog.csdn.net/ncc1995/article/details/99456645 ...
评论 1
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值