pytorch加载模型时报错 AttributeError: Can‘t get attribute ‘CNN‘ on ＜module ‘__main__‘＞

最新推荐文章于 2024-08-24 15:21:05 发布
最新推荐文章于 2024-08-24 15:21:05 发布
阅读量1.6k 收藏 2
点赞数 3
分类专栏： bug笔记 文章标签： pytorch cnn 深度学习
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/qq_43750528/article/details/130295800
版权

1、问题描述

在Pytorch加载模型时报错：
AttributeError: Can’t get attribute ‘CNN’ on <module ‘__ main __’ from 'E:\…

看了很多篇其他人的分享，都比较复杂或者不适合我这种情况。

在这里插入图片描述

2、解决办法

将你之前训练模型时，定义模型的类全部复制到这个py文件中即可

对于我这种情况，即包括__init__方法和forward方法的整个CNN网络，如下：

在这里插入图片描述

蜡笔大新home
关注
  • 3
    点赞
  • 踩
  • 2
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
专栏目录
Python多线程报错：on ＜module ‘__main__‘ (built-in)＞
weixin_39862968的博客
07-21 5194
python多线程填坑：报错 Attribute Error: Can’t get attribute ‘你的多线程处理函数’ on <module ‘main’ (built-in)> jupyter 使用多线程时 from multiprocessing import Pool def down_temp(data): print(i) ... pool=Pool() for data in range(10): pool.apply_async(func=down_temp,args=
载入模型错误：AttributeError: Can‘t get attribute ‘CNN‘ on ＜module ‘__main__‘ from XXXX
weixin_59995798的博客
01-03 857
AttributeError: Can't get attribute 'CNN' on <module '__main__' from XXX 这个错误是由于保存模型中，有CNN类的序列化导致的，我在网上也看了好多教程，方法各种各样（但有的不可行），总结一下： （本人没有系统学过python，属于边用边学，有错误的话欢迎指正，序列化的话我在Java中学过，但相信原理都是相通的） 先说一下，我出现这个原因的时候，CNN构建的写法： 1、我在cnn_model.py文件中构建了一个CNN类，
AttributeError: Can't get attribute 'Net' on module '__main__'
热门推荐
baidu_36639782的博客
01-31 3万+
在使用pytorch加载模型时报错： torch.save(old_model, PATH) new_model = torch.load(PATH) AttributeError: Can't get attribute 'Net' on &lt;module '__main__'&gt;   解决办法： 1、将类的定义添加到加载模型的这个py文件中，这个方法有点。。。 2、使用官...
解决Can‘t get attribute ‘CNN‘ on ＜module ‘__main__‘＞
李不理理的博客
01-23 831
解决Can't get attribute 'CNN' on
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
Pytorch中torch.load()中出现AttributeError: Can‘t get attribute
wangchaoxjtu的专栏
05-17 4206
Pytorch中torch.load()中出现AttributeError: Can't get attribute 原因：保存下来的模型和参数不能在没有类定义时直接使用。 Pytorch使用Pickle来处理保存/加载模型，这个问题实际上是Pickle的问题，而不是Pytorch。 解决方法也非常简单，只需显式地导入类定义。即将包含类定义的文件复制粘贴到与要运行的文件同一文件夹下，再import Class！ 但是，在实际过程中，我们采取import Class的方法没起到作用，而直接在要运行的文
8.27(cnn example).rar_CNN_CNN pytorch_python cnn_pytorch_pytorch
07-14
python实现cnn简单例子，pytorch框架
PytorchExample-master.zip_CNN_I78E_pytorch_pytorch CNN_自定义数据集
07-15
4. **构建CNN模型**：PyTorch中的`nn.Module`是所有神经网络层和模型的基类。你需要创建一个子类，定义网络结构，包括卷积层（`nn.Conv2d`）、池化层（`nn.MaxPool2d`）、全连接层（`nn.Linear`）等，并重写`forward...
deep_learing_pytorch.zip_CNN_MNIST pytorch_pytorch_优化CNN_卷积神经网络
07-15
在本项目中，我们探索了如何使用PyTorch框架构建并优化一个卷积神经网络（CNN）模型，以解决MNIST手写数字识别任务。MNIST数据集是机器学习领域中一个经典的数据集，用于训练和测试图像分类算法。下面我们将深入讨论...
Person_reID_baseline_pytorch-master.zip_Person Re-ID:_PyTorch行人识
09-20
本项目基于PyTorch框架，提供了行人重识别的基线模型，其在Rank-1评估指标上达到了88%的准确率，而在平均精度均值（Mean Average Precision，mAP）上则达到了70%的性能，显示出该模型在实际应用中的高效性和准确性。...
BigGAN.zip_image processing_pytorch cnn
09-23
标题中的“BigGAN.zip_image processing_pytorch cnn”揭示了我们关注的主题是关于图像处理的，具体来说，是利用PyTorch框架实现的卷积神经网络（CNN）模型——BigGAN。BigGAN是一种深度学习模型，它在生成对抗网络...
pytorch 在加载模型时出现的错误
xixilidehuoshan的博客
01-02 5908
在训练完后，在另一个.py中想导入模型进行使用。导入之后出现这样的错误： pytorch AttributeError: Can’t get attribute ‘Net1’ on &amp;lt;module ‘main’&amp;gt; 网上有很多原因， 有一部分是说多线程。 但经过查看文档，这种情况不属于多线程的问题。 是因为在保存模型时pytorch保存整个模型的序列化数据绑定到特定类以及保存模型时...
AttributeError Can‘t get attribute ‘Model‘ on module ‘main‘
Akun_2217的博客
12-01 596
torch.load时会出现这种错误，本质是在当前文件内运行时并没有发现Model类。AttributeError经常是说明调用了某一个类没有的属性。
Pytorch保存和加载训练好的模型，以及加载中提示Can‘t get attribute ‘XXX‘ on ＜module ‘__main__‘ from ‘YYY‘＞的解决方法
weixin_45779073的博客
03-11 4020
保存，分仅保存参数，和保存整个模型 1 仅保存参数，轻便，但需把网络模型先执行一遍 方法百度 2 保存整个模型，无脑全存 保存 torch.save(rnn, R'D:\Python\11111.pkl') 加载. rnn = torch.load(R'D:\Python\11111.pkl') 如果直接执行，报错 AttributeError: Can't get attribute 'RNN' on <module '__main__' from '<input&gt
【PyTorch】AttributeError: Can't get attribute 'BCNN'
ncc1995的博客
08-13 8531
错误：AttributeError: Can't get attribute 'BCNN' on <module '__main__' from ' 今天在服务器上训练好一个网络，用torch.save(model, 'net.pkl')将模型和参数一并保存了起来，方便在自己电脑上测试。 把保存的模型下载下来进行测试的时候发现报了上述错误。 原来保存下来的模型和参数不能在没有类定义时直...
torch.load时报AttributeError: Can‘t get attribute ‘Module‘ on ＜module ‘__main__‘错误的解决方法
人工智能
05-11 2781
这里就是由于我们在test_1.py里定义了Module这个类，在序列化的时候也把这个类序列化进去了，当我们在另一个文件中加载的时候，由于没有Module这个类在内存里，就会导致找不到Module这个模型类进行反序列化成对象。torch.save()函数使用的是google的protobuf协议进行序列化的，当我们在save一个已经存在的Module的时候，在另一个py文件中load是没有问题的。的时候，问题也是一样，按照上面的方式解决就行！这样就可以正常运行了。
【工程实践】“AttributeError: Can‘t get attribute ‘BertClassificationModel‘ on ＜module ‘__main__‘＞“加载模型报错
Dong_ZH的博客
05-08 2413
1.问题描述 将训练好的模型使用 torch.save(model, path)命令进行保存后，通过Trained_model = torch.load(path)命令进行加载，常出现错误。具体代码和出现错误详情如下： import torch import torch.nn as nn #引进训练好的模型进行测试 path = '/root/Save_model/bert_hide_model.pkl' #模型加载 Trained_model = torch.load(path) 出现错误：
爆改YOLOv8 | 利用MB-TaylorFormer提高YOLOv8图像去雾检测
最新发布
weixin_43986124的博客
08-24 530
python，yolo，yolov8，目标检测，暴力涨点
pytorchCan't get attribute 'Model' on <module '__main__'>
07-27
The error message "Can't get attribute 'Model' on <module '__main__'>" usually occurs when you try to access an attribute or class that does not exist in the current module. In this case, it seems like you are trying to access the 'Model' attribute but it is not available. To resolve this issue, make sure that you have imported the necessary modules and classes correctly. In the case of PyTorch, you need to import the specific module or class that contains the 'Model' attribute. For example, if you are trying to use the PyTorch model class, you should import it like this: ```python import torch import torch.nn as nn model = nn.Model() # Create an instance of the model ``` Make sure that you have installed PyTorch correctly and that you are using the correct syntax to access the desired attribute or class.
评论 1
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值