PyTorch模型保存与加载

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

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

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

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

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

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

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

 2. torch.save / torch.load

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

 功能:保存对象到硬盘中

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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值