解决方法

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

报错原因

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

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐