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

最新推荐文章于 2024-08-29 19:29:50 发布

xixilidehuoshan

最新推荐文章于 2024-08-29 19:29:50 发布

阅读量5.9k

点赞数 7

分类专栏：机器学习实战文章标签： pytorch 导入模型

本文链接：https://blog.csdn.net/xixilidehuoshan/article/details/85636918

版权

机器学习实战专栏收录该内容

3 篇文章 0 订阅

订阅专栏

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