在使用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实现的一些基本的分析代码。欢迎大家关注，二维码如下：

相关文章列表如下：

 

# 机器学习 # 深度学习 # pytorch
Logo
MCP技术社区

欢迎加入 MCP 技术社区！与志同道合者携手前行，一同解锁 MCP 技术的无限可能！

更多推荐

cover

8种封装的1700V国产碳化硅(SiC)功率模块产品介绍及应用

avatar MCP技术社区
cover

（一篇入门）汽车电子电器 之 电机MCU控制器 四

avatar MCP技术社区

OpenCV C++ 入门实战：从基础操作到类封装全解析

Mat数据结构、图像读写、颜色空间转换、像素操作、算术运算和轨迹栏交互，以及面向对象封装思想。图像处理：学习滤波（高斯滤波、中值滤波）、边缘检测（Canny、Sobel）、形态学操作（腐蚀、膨胀）。目标检测：掌握轮廓提取（）、特征匹配（）、Haar 级联分类器。视频处理：使用读取视频，对帧进行实时处理。性能优化：学习多线程、GPU 加速（cv::cuda模块），提升处理速度。通过持续实践，可逐步掌

avatar MCP技术社区