pytorch加载模型时报错：AttributeError: Can‘t get attribute ‘MobileNet‘ on 「module ‘main‘

最新推荐文章于 2024-04-27 18:29:43 发布

dataat

最新推荐文章于 2024-04-27 18:29:43 发布

阅读量1.6w

点赞数 8

分类专栏： python pytorch 文章标签：机器学习深度学习 pytorch

本文链接：https://blog.csdn.net/booklijian/article/details/107177308

版权

python 同时被 2 个专栏收录

46 篇文章 1 订阅

订阅专栏

pytorch

19 篇文章 0 订阅

订阅专栏

在使用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-packages\torch\serialization.py", line 426, in load
    return _load(f, map_location, pickle_module, **pickle_load_args)
  File "D:\Python36\lib\site-packages\torch\serialization.py", line 613, in _load
    result = unpickler.load()
AttributeError: Can't get attribute 'MobileNet' on <module '__main__' from 'model2android.py'>

Process finished with exit code 1

查看了一下源代码，其中定义了一个类MobileNet，因此把该类的定义代码全部复制到新的python程序中，再次运行，顺利通过。实际上也可以将MobileNet的定义单独找一个文件中，然后import到新的python同样也可以。

-------------------- 正文到此结束------------------------

推荐一个公众号：健哥聊量化，会持续推出股票相关基础知识，以及python实现的一些基本的分析代码。欢迎大家关注，二维码如下：

相关文章列表如下：

dataat

关注

8
点赞
踩
7

收藏

觉得还不错? 一键收藏
2
评论
pytorch加载模型时报错：AttributeError: Can‘t get attribute ‘MobileNet‘ on 「module ‘__main__‘

在使用pytorch加载模型时(具体命令为：torch.load("xxxxx.ckpt"),报错如下：D:\Python36\python.exe model2android.pyTraceback (most recent call last): File "model2android.py", line 3, in <module> model=torch.load("MobileNetv2.cpkt") File "D:\Python36\lib\site-pac
复制链接

扫一扫