当前位置: 首页 > news >正文

PyTorch模型保存与加载

1.保存与加载的概念(序列化与反序列化)

模型训练完毕之后,肯定想要把它保存下来,供以后使用,不需要再次去训练。

那么在pytorch中如何把训练好的模型,保存,保存之后又如何加载呢?

这就用需要序列化与反序列化,序列化与反序列化的概念如下图所示:

        因为在内存中的数据,运行结束会进行释放,所以我们需要将数据保存到硬盘中,以二进制序列的形式进行长久存储, 便于日后使用。

        序列化即把对象转换为字节序列的过程,反序列化则把字节序列恢复为对象。

         在pytorch中,对象就是模型,所以我们常常听到序列化和反序列化,就是将训练好的模型从内存中保存到硬盘里,当要 使用的时候,再从硬盘中加载。

 2. torch.save / torch.load

pytorch提供的序列化与反序列化函数分别是

 功能:保存对象到硬盘中

主要参数: obj- 对象 f - 文件路径

http://www.lryc.cn/news/472462.html

相关文章:

  • CH569开发前的测试
  • MySQL中表的外连接和内连接
  • Ubuntu 上安装 Redmine 5.1 指南
  • 从变量的角度理解 Hooks , 变得更简单了
  • LabVIEW Modbus通讯稳定性提升
  • (8) cuda分析工具
  • C语言 | Leetcode C语言题解之第517题超级洗衣机
  • Java多线程编程基础
  • 刷代随有感(134):单调栈——下一个更大元素I(难点涉及哈希表与单调栈的结合)
  • Linux云计算 |【第五阶段】CLOUD-DAY5
  • 被上传文件于后端的命名策略
  • 哈希表 算法专题
  • unity3d————[HideInInspector]
  • Soanrquber集成Gitlab 之 导入Gitlab项目
  • 论区块链技术及应用
  • GPT避坑指南:如何辨别逆向、AZ、OpenAI官转
  • Qt 文本文件读写与保存
  • Linux基础环境搭建(CentOS7)- 安装Scala和Spark
  • SpringBoot 下的Excel文件损坏与内容乱码问题
  • 官宣下代GPU存在缺陷,50系显卡或将迎来涨价
  • 使用pytorch实现LSTM预测交通流
  • C/C++(八)C++11
  • 使用three.js 实现 自定义绘制平面的效果
  • 玩转Docker | 使用Docker部署捕鱼网页小游戏
  • 第2章 Android App开发基础
  • 通过 SYSENTER/SYSEXIT指令来学习系统调用
  • Nginx开发实战——网络通信(一)
  • w外链如何跳转微信小程序
  • 获取平台Redis各项性能指标
  • STM32 HAL 点灯