代码拉取完成,页面将自动刷新
译者:yuange250
Pytorch主要有两种方法可用于序列化和保存一个模型。
第一种只存取模型的参数(更为推荐): 保存参数:
torch.save(the_model.state_dict(), PATH)
读取参数:
the_model = TheModelClass(*args, **kwargs)
the_model.load_state_dict(torch.load(PATH))
第二种方法则将整个模型都保存下来:
torch.save(the_model, PATH)
读取的时候也是读取整个模型:
the_model = torch.load(PATH)
在第二种方法中, 由于特定的序列化的数据与其特定的类别(class)相绑定,并且在序列化的时候使用了固定的目录结构,所以在很多情况下,如在其他的一些项目中使用,或者代码进行了较大的重构的时候,很容易出现问题。
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。