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

解决方法

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

问题解析

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

你可能感兴趣的:(python,python)