pytorch 在加载模型时出现的错误

时间:2024-03-29 11:18:30

在训练完后,在另一个.py中想导入模型进行使用。导入之后出现这样的错误:
pytorch AttributeError: Can’t get attribute ‘Net1’ on <module ‘main’>
pytorch 在加载模型时出现的错误
网上有很多原因, 有一部分是说多线程。 但经过查看文档,这种情况不属于多线程的问题。 是因为在保存模型时pytorch保存整个模型的序列化数据绑定到特定类以及保存模型时使用的确切目录结构。 所以当你在另一个项目中导入会发生错误,或者中断。
解决的方式很简单,就是将声明模型的class的部分代码或者你模型定义的代码加进新的项目中,这样就可以正常加载使用了。

相关文章