pytorch模型存储的2种实现方法
2020-03-11 16:12
1131 查看
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种实现方法就是小编分享给大家的全部内容了,希望能给大家一个参考
您可能感兴趣的文章:
相关文章推荐
- 常用的几种交互表存储过程的实现方法
- mssql server 存储过程里,bulk insert table from '路径+文件',路径固定,文件名不固定的实现方法
- WindowsXP下通过硬盘安装Ubuntu实现双系统2种方法对比分析
- PHP实现MVC开发得最简单的方法――模型
- pl/sql存储过程中游标嵌套的实现方法
- 基于线程实现的生产者消费者模型(Object.wait(),Object.notify()方法)
- [java]多线程2种实现方法
- C++接口与实现分离的2种方法
- 常用的几种交互表存储过程的实现方法
- 2种简单的Apache+PHP下实现URL重写的方法
- 基于AVI的网络视频监控存储系统的实现方法
- android手机实现拍照的2种方法
- ExtJS与PHP、MySQL实现存储的方法
- Android中ImageButton自定义按钮的按下效果的代码实现方法,附网上2种经典解决方法。
- 新旧MFC版本CEdit透明的2种实现方法
- (轉貼) C++接口与实现分离的2种方法 (C/C++)
- 常用的几种交互表存储过程的实现方法
- INFORMIX存储过程中实现排序后结果insert到表的方法
- 常用的几种交互表存储过程的实现方法
- ACCESS 调用后台存储过程的实现方法