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

在使用pytorch加载模型时(具体命令为：torch.load("xxxxx.ckpt"),报错如下：

D:\Python36\python.exe model2android.py
Traceback (most recent call last):
  File "model2android.py", line 3, in 
    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 

Process finished with exit code 1

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

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

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

相关文章列表如下：

• 股票基础知识----- K线形态

• 股票K线形态 ----早晨之星

• “早晨之星”实际操作篇---通达信软件为例

• 牛刀小试----python+tushare进行股票分析

• 股票K线形态----黄昏之星

• 股票K线形态-----墓碑线

• 股票K线形态-----多方炮

• 股票K线形态-----红三兵

• 股票K线形态----三只乌鸦

• 股票K线形态-----锤头线、吊颈线、倒锤头线