今天小编就为大家分享一篇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种实现方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持爱安网。

最新资讯
荣耀V40系列搭载联发科天玑1000+处理器 售价3599元起

荣耀V40系列搭载联发

荣耀今日举行V40系列新品发布会,这也是荣耀独立后的首
刚收购完王思聪的香蕉游戏,英雄体育VSPN再获6000万美元融资

刚收购完王思聪的香蕉

1月22日,英雄体育VSPN宣布完成6000万美元B+轮融资,投资
对微信8.0版本的非官方解读

对微信8.0版本的非官

这篇解读,角度会和其他介绍文章的关注点不太一样,希望对
蔚来宣布恢复免费换电政策:每月可享受4-6次

蔚来宣布恢复免费换电

蔚来今天宣布调整用户终身免费换电权益。
欧菲光将出(bu)售(mai)华南厂?传闻引发苹果概念股集体大跌

欧菲光将出(bu)售(mai)华南厂?传

1月21日有媒体报道,数位知情人士透露,欧菲光华南厂(欧菲
荣耀赵明:荣耀有8000名员工,其中50%是研发人员

荣耀赵明:荣耀有8000名

1月22日,荣耀举行独立后的首个新品发布会。CEO赵明在发
最新文章
在pycharm中为项目导入anacodna环境的操作方法

在pycharm中为项目导

这篇文章主要介绍了在pycharm中为项目导入anacodna环
tensorflow的ckpt及pb模型持久化方式及转化详解

tensorflow的ckpt及pb

今天小编就为大家分享一篇tensorflow的ckpt及pb模型持
PyTorch笔记之scatter()函数的使用

PyTorch笔记之scatter

这篇文章主要介绍了PyTorch笔记之scatter()函数的使用
python3实现网页版raspberry pi(树莓派)小车控制

python3实现网页版ras

这篇文章主要为大家详细介绍了python3实现网页版raspb
完美解决pycharm导入自己写的py文件爆红问题

完美解决pycharm导入

今天小编就为大家分享一篇完美解决pycharm导入自己写
pycharm内无法import已安装的模块问题解决

pycharm内无法import

今天小编就为大家分享一篇pycharm内无法import已安装