pytorch模型存储的2种实现方法

 更新时间:2020年02月14日 12:00:57   作者:慢行厚积   我要评论
今天小编就为大家分享一篇pytorch模型存储的2种实现方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧

1、保存整个网络结构信息和模型参数信息:

torch.save(model_object, './model.pth')

直接加载即可使用:

model = torch.load('./model.pth')

2、只保存网络的模型参数-推荐使用

torch.save(model_object.state_dict(), './params.pth')

加载则要先从本地网络模块导入网络,然后再加载参数:

from models import AgeModel
model = AgeModel()
model.load_state_dict(torch.load('./params.pth'))

以上这篇pytorch模型存储的2种实现方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持脚本之家。

相关文章

  • Python 实现一行输入多个值的方法

    Python 实现一行输入多个值的方法

    下面小编就为大家分享一篇Python 实现一行输入多个值的方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2018-04-04
  • 解决python3 安装不了PIL的问题

    解决python3 安装不了PIL的问题

    今天小编就为大家分享一篇解决python3 安装不了PIL的问题,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2019-08-08
  • HTML的form表单和django的form表单

    HTML的form表单和django的form表单

    这篇文章主要介绍了HTML的form表单和django的form表单,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2019-07-07
  • PyCharm下载和安装详细步骤

    PyCharm下载和安装详细步骤

    这篇文章主要介绍了PyCharm下载和安装详细步骤,本文图文并茂给大家介绍的非常详细,具有一定的参考借鉴价值,需要的朋友可以参考下
    2019-12-12
  • Python连接Redis的基本配置方法

    Python连接Redis的基本配置方法

    本文分享了Linux系统下Python连接Redis的基本配置方法,Python连接Redis需要Redis的Python包支持
    2018-09-09
  • Python环境下安装使用异步任务队列包Celery的基础教程

    Python环境下安装使用异步任务队列包Celery的基础教程

    这篇文章主要介绍了Python环境下安装使用异步任务队列包Celery的基础教程,Celery的分布式任务管理适合用于服务器集群的管理和维护,需要的朋友可以参考下
    2016-05-05
  • python 自动化将markdown文件转成html文件的方法

    python 自动化将markdown文件转成html文件的方法

    这篇文章主要介绍了python 自动化将markdown文件转成html文件的方法的相关资料,本文介绍的非常详细,具有参考借鉴价值,需要的朋友可以参考下
    2016-09-09
  • python多行字符串拼接使用小括号的方法

    python多行字符串拼接使用小括号的方法

    今天小编就为大家分享一篇python多行字符串拼接使用小括号的方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2018-07-07
  • Python实现时间序列可视化的方法

    Python实现时间序列可视化的方法

    matplotlib库是一个用于创建出版质量图表的桌面绘图包(2D绘图库),是Python中最基本的可视化工具。这篇文章主要介绍了Python时间序列可视化实现,需要的朋友可以参考下
    2019-08-08
  • python实现时间o(1)的最小栈的实例代码

    python实现时间o(1)的最小栈的实例代码

    这篇文章主要介绍了python实现时间o(1)的最小栈的实例代码,小编觉得挺不错的,现在分享给大家,也给大家做个参考。一起跟随小编过来看看吧
    2018-07-07

最新评论