当前位置:   article > 正文

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

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博客】
推荐阅读
相关标签
  

闽ICP备14008679号