深度学习文件读写

本文介绍了如何使用torch.save和torch.load在PyTorch中保存和加载张量,包括张量列表和神经网络模型的state_dict。同时解释了eval方法在深度学习中用于设置模型为评估模式,避免训练时的梯度计算和参数更新,仅用于预测阶段。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

加载与保存张量使用torch.save(要保存的张量,‘名称’),torch.load('名称')

张量列表与字典均可,以下举例张量列表

import torch
x=torch.arange(4)
y=torch.zeros(4)
torch.save([x,y],'xy_list')
xx,yy=torch.load('xy_list')

对于深度学习神经网络,为了保存所有的参数,采用以下代码:

#假设前面定义了一个网络net=MLP()
torch.save(net.state_dict(),'statedict')
#为了恢复模型,进行备份
clone_net=MLP()
clone_net.load_state_dict(torch.load('statedict'))
clone_net.eval()
#此时clone_net与原本net恢复一致

eval是做什么的?在深度学习中,调用eval()方法是为了将模型设置为评估模式,即告诉模型在推理(inference)阶段运行时不要进行梯度计算和参数更新。这通常用于在模型训练完成后,进行模型在测试集或实际应用中的预测时使用。

load_state_dict是将模型恢复原本参数的内置函数

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值