torch保存、读取模型_torch moudle 从模型中读取属性

作者：你好赵伟 | 2024-06-30 13:28:00

踩

torch moudle 从模型中读取属性

方法一:

将整个模型保存、读取:

torch.save(model, PATH)
model = torch.load(PATH)
1
2

仅保存模型的state_dict()、读取:

torch.save(model.state_dict(), PATH)
state_dict = torch.load(PATH)
1
2

这时仅读取了state_dict，还需要:

new_model.load_state_dict(torch.load(PATH)) # 其中的new_model是新初始化的模型
1

靠，直接存模型翻车了，爆了类似于在这里插入图片描述这样的错误，以后还是只保存state_dict吧。

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/你好赵伟/article/detail/772753