解决pytouch导入模型报错:AttributeError: Can‘t get attribute ‘XXX‘ on <module ‘__main__‘ from XXX>

  • Post author:
  • Post category:其他

解决方法

看到这个报错的文件位置:<module '__main__' from XXX,只需要把自定义的那个模型的类,即Can't get attribute 'XXX'这里的XXX,直接把这个类复制到文件位置这里即可。

问题解析

使用pytouch导入模型的时候有一个pickle的操作,但是因为未知自定义的模型的结构,所以无法解析模型