pytorch模型保存和加载

模型保存

torch.save()实现对网络结构和模型参数的保存.有两种保存方式:一是保存整个神经网络的结构信息和模型参数信息.save的对象是网络net.二是只保留神经网络的训练模型参数,save的对象是net.state_dict()

torch.save('net1','model.pkl') #保留整个神经网络的结构和模型参数
torch.save(net1.state_dict(),'model.pkl') # 只保留神经网络的模型参数

模型加载

对于两种保存方式,重载也有两种方式.
对应第一种完整网络结构信息,重载的时候通过

torch.load('model.pkl')

直接初始化新的神经网络对象即可.
对应第二种只保存模型参数信息,需要首先导入对应的网络,通过

net.load_state_dict(torch.load('model.pkl'))

完成模型参数的重载
在网络比较大时,第一种方法会花费较多的时间.

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

  • 搭建ConvNet 所有的网络都要继承torch.nn.Module,然后在构造函数中使用torch.nn中的提供...
    DeepWeaver阅读 6,251评论 0 0
  • 6.23 翻译了发出的信函 翻译了报验的资料 协助专业工程师修改材料清单
    prince哈哈阅读 1,463评论 0 0
  • 我家附近的一个水塘。 我拿出了一根长棍子,上面连着一起两个网兜,一个大一点在上一个小一点紧挨着大的,有点重叠,感觉...
    拾小玖阅读 1,445评论 0 0
  • 母亲:孩子,你能不能戒掉游戏?学习才是你的正经事。 孩子:妈妈,你能不能戒掉麻将?做饭,收拾屋子才是你的正经事。 ...
    泰山寒梅阅读 3,950评论 50 49
  • ⒈ 明月近几年经常换工作。这不,七月底又来到朋友斌任总经理的中山一家公司上班。 过来之前,斌一直说这边跟他之前的工...
    明月牵你阅读 3,157评论 13 11

友情链接更多精彩内容