码迷,mamicode.com
首页 > 其他好文 > 详细

Pytorch加载和保存模型

时间:2019-09-10 14:45:21      阅读:112      评论:0      收藏:0      [点我收藏+]

标签:pat   反序列化   div   方法   tps   pre   path   shu   文档   

首先注明,参考了这篇博客https://www.jianshu.com/p/4905bf8e06e5

  • 方法1、

先序列化,格式可以是mdl,pt等

torch.save(model.state_dict(), MODEL_PATH)

然后反序列化,再加载

model.load_state_dict(torch.load(MODEL_PATH))
  • 方法2、

保存整个模型,格式可以是pth.tar

 torch.save(model, PATH)

然后加载

model = torch.load(PATH)

还有一些在GPU和CPU之间的传输方法,具体见博客或者官方文档

Pytorch加载和保存模型

标签:pat   反序列化   div   方法   tps   pre   path   shu   文档   

原文地址:https://www.cnblogs.com/yqpy/p/11497259.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!