这个屌丝很懒，什么也没留下！

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博客】