解决pytorch导入模型报错：AttributeError: Can‘t get attribute ‘XXX‘ on ＜module ‘__main__‘ from YYY＞

解决方法

看到这个报错的文件位置：main’ from YYY，只需要把自定义的那个模型的类，即Can’t get attribute ‘XXX’ 这里的 XXX，直接把这个类复制到文件位置 YYY 这里即可。

报错原因

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