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

在训练完后，在另一个.py中想导入模型进行使用。导入之后出现这样的错误： pytorch AttributeError: Can’t get attribute ‘Net1’ on <module ‘main’> 网上有很多原因， 有一部分是说多线程。 但经过查看文档，这种情况不属于多线程的问题。 是因为在保存模型时pytorch保存整个模型的序列化数据绑定到特定类以及保存模型时使用的确切目录结构

>>阅读原文<<