PyTorch 1.0 中文文档：序列化的相关语义

ApacheCN_飞龙

发布于 2022-05-07 14:12:01

1470

发布于 2022-05-07 14:12:01

文章被收录于专栏：信数据得永生

译者：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)

阅读全文／改进本文

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2019-02-17，如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自作者个人站点/博客前往查看

如有侵权，请联系 cloudcommunity@tencent.com 删除。

本文参与腾讯云自媒体同步曝光计划，欢迎热爱写作的你一起参与！

登录后参与评论

0 条评论

热度