pytorch：使用pytorch加载模型时报错

最新推荐文章于 2024-01-08 14:07:25 发布
最新推荐文章于 2024-01-08 14:07:25 发布
阅读量1k 收藏
点赞数
分类专栏： 后端 Python 文章标签： python 人工智能 pytorch
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_38676276/article/details/113338778
版权

在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

乐事原味~
关注
  • 0
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
Pytorch加载模型时报错
carlwuuu的博客
12-28 1732
加载模型时出现Unexpected key(s) in state_dict错误 报错截图如下： 反复排查问题没发现为何如此，查看pytorch中文文档发现保存和加载模型方法都完全正确，模型保存和加载代码对比中文文档截图如下： 其中一个方法是在加载模型时添加参数strict=False，可以只保留键值相同的参数避免出错，用法如下：model.load_state_dict(ckpt[‘stat...
pytorch加载模型时报错
qq_39076073的博客
06-03 326
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时报AttributeError: Can‘t get attribute ‘Module‘ on ＜module ‘__main__‘错误的解决方法
人工智能
05-11 2778
这里就是由于我们在test_1.py里定义了Module这个类，在序列化的时候也把这个类序列化进去了，当我们在另一个文件中加载的时候，由于没有Module这个类在内存里，就会导致找不到Module这个模型类进行反序列化成对象。torch.save()函数使用的是google的protobuf协议进行序列化的，当我们在save一个已经存在的Module的时候，在另一个py文件中load是没有问题的。的时候，问题也是一样，按照上面的方式解决就行！这样就可以正常运行了。
AttributeError: Can‘t get attribute ‘xxx‘ on ＜module ‘__main__‘ from ‘xxx‘
I_am_programmer的博客
04-17 4271
在学习python时，开发学生选课系统，期间出现AttributeError: Can't get attribute 'Course' on 报错。然后我们看下报错代码，当调用get_from_pickle读取student_info或者course_info时，就会出现。
AttributeError Can‘t get attribute ‘Model‘ on module ‘main‘
Akun_2217的博客
12-01 595
torch.load时会出现这种错误，本质是在当前文件内运行时并没有发现Model类。AttributeError经常是说明调用了某一个类没有的属性。
pytorch加载模型时报错：AttributeError: Can‘t get attribute ‘MobileNet‘ on 「module ‘__main__‘
booklijian的专栏
07-07 1万+
在使用pytorch加载模型时(具体命令为：torch.load("xxxxx.ckpt"),报错如下： D:\Python36\python.exe model2android.py Traceback (most recent call last): File "model2android.py", line 3, in <module> model=torch.load("MobileNetv2.cpkt") File "D:\Python36\lib\site-pac
解决“”AttributeError: Can‘t get attribute ‘Model‘ on
奥利波德的魔法屋
04-08 9668
相关解释 解决方案： 将原模型的类文件导入 如下（即将原网络的class文件导入）
flask_pytorch：使用flask运行pytorch模型
02-05
flask_pytorch 使用烧瓶运行pytorch模型启动服务器python3 app.py 您将看到这样的结果向pytorch服务器提交请求python3 request_sample.py -f= ' file_path ' 发送这样的图像 您会看到这样的结果（我以mnist图像为例...
SCNN-pytorch:PyTorch中的空间CNN模型使用Cityscapes-dataset
03-18
pytorch-SCNN **于2018/12/15更新。 去做 支持不同的骨干网 支持城市景观 单GPU训练 介绍 这是的PyTorch（0.4.1）实现。它可以使用Modified Aligned ResNet作为主干。目前，我们使用Cityscapes数据集训练SCNN。 ...
pytorch：pytorch模型训练的主要步骤
07-27
代码中主要包含pytorch模型训练的主要步骤，可以根据需要自定义模型、数据集，并进行训练。
pytorch：表情识别模型ResNet
最新发布
07-27
使用PyTorch来训练一个基于ResNet的表情识别网络
深度学习框架PyTorch：入门与实践_高清1
08-04
《深度学习框架PyTorch：入门与实践》是一本针对PyTorch这一深度学习框架的教程，旨在帮助读者快速掌握PyTorch的基础知识并实践应用。PyTorch因其简洁的设计、直观的接口、高效的执行速度和高度的灵活性，自2017年...
解决pytouch导入模型torch.load报错：AttributeError: Can‘t get attribute ‘XXX‘ on ＜module ‘__main__‘ from XXX＞
levelina的博客
01-02 1227
解决办法：在torch.load（）之前，加入你定义的模型的CLASS，也就是把训练模型时的定义的模型的类复制到torch.load（）之前。
Can‘t get attribute ‘_rebuild_tensor_v2‘ on module ‘torch._utils‘ 解读
牛肉胡辣汤
01-08 997
当你在使用PyTorch时遇到"Can't get attribute '_rebuild_tensor_v2' on module 'torch._utils'"错误时，这通常是由于不兼容的PyTorch版本引起的。通过更新PyTorch版本、检查代码兼容性、调整PyTorch依赖或重新安装PyTorch，你可以解决这个问题并让你的代码重新正常运行。
解决pytorch导入模型报错：AttributeError: Can‘t get attribute ‘XXX‘ on ＜module ‘__main__‘ from YYY＞
MuMengSunny
07-07 7337
pytorch导入模型权重报错
Python：实现蓝牙通信
热门推荐
Jcsim~
01-23 2万+
感谢：（1）win10安装python蓝牙依赖https://blog.csdn.net/caigen0001/article/details/110877401 （2）https://blog.csdn.net/weixin_50396804/article/details/109823229 https://www.cnblogs.com/-wenli/p/11646928.html 一、蓝牙通信流程 和java版的通信流程一样（j...
Python：如何向字典里面添加数据
Jcsim~
07-22 1万+
直接赋值即可 data= {'name':'AI','age':'1024'} #执行以下代码 data['update_date'] = '2020-10-24' print(data) #输出结果 {'name':'AI','age':'1024','update_date':'2020-10-24'}
org.pytorch:pytorch_android:1.12.1
04-29
org.pytorch:pytorch_android:1.12.1是一个用于在Android平台上运行PyTorch模型的库。它提供了一组API和工具，使开发者能够在Android设备上加载、运行和推理PyTorch模型。 具体来说，org.pytorch:pytorch_android:1.12.1库包含以下主要功能： 1. 模型加载和推理：该库允许开发者将训练好的PyTorch模型加载到Android设备上，并使用它们进行推理。开发者可以使用Java或JNI接口来调用模型，并获取模型的输出结果。 2. 模型转换和优化：该库提供了工具和API，用于将PyTorch模型转换为适用于Android设备的格式，并进行优化以提高推理性能。这样可以确保模型在移动设备上的高效运行。 3. 模型部署和集成：该库支持将PyTorch模型与Android应用程序集成，使开发者能够在移动设备上直接使用模型进行推理。这样可以实现离线推理、实时推理等应用场景。
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值