torch.save()

保存为pt或者pth都可以,前提似乎都得先mkdir或者makedirs,不能直接save到一个即时指定的文件夹下。

pytorch保存模型等相关参数,利用torch.save(),以及读取保存之后的文件
本文分为两部分,第一部分讲如何保存模型参数,优化器参数等等,第二部分则讲如何读取。

假设网络为model = Net(), optimizer = optim.Adam(model.parameters(), lr=args.lr), 假设在某个epoch,我们要保存模型参数,优化器参数以及epoch

一、save

  1. 先建立一个字典,保存三个参数:

state = {‘net’:model.state_dict(), ‘optimizer’:optimizer.state_dict(), ‘epoch’:epoch}

2.调用torch.save():torch.save(state, dir)

其中dir表示保存文件的绝对路径+保存文件名,如’/home/qinying/Desktop/modelpara.pth’

二、load

当你想恢复某一阶段的训练(或者进行测试)时,那么就可以读取之前保存的网络模型参数等。

checkpoint = torch.load(dir)

model.load_state_dict(checkpoint[‘net’])

optimizer.load_state_dict(checkpoint[‘optimizer’])

start_epoch = checkpoint[‘epoch’] + 1

更多参考

https://www.cnblogs.com/qinduanyinghua/p/9311410.html
https://blog.csdn.net/weixin_38410551/article/details/101535705
https://blog.csdn.net/weixin_38145317/article/details/111152491

  • 7
    点赞
  • 22
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
torch.save和torch.jit是PyTorch中用于保存和转化模型的工具。 torch.save用于将PyTorch模型保存到磁盘上,以便以后可以重新加载和使用。它有三种使用场景,具体可以参考。 torch.jit是PyTorch的即时编译工具,用于将PyTorch模型转化为可序列化和优化的格式。torch.jit.trace用于将一个具有输入和输出的模型转化为脚本模型,而torch.jit.script用于将一个具有复杂控制流的模型转化为脚本模型。这些转换后的模型可以在不依赖Python运行时的情况下运行,并且可以实现更好的性能和部署。具体可以参考和。 总结起来,torch.save用于保存模型,而torch.jit用于将模型转化为可序列化和优化的格式,以提高性能和部署效果。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* [.torch.save与torch.jit.save](https://blog.csdn.net/jianminfly/article/details/125387387)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"] - *2* *3* [TorchScript (将动态图转为静态图)(模型部署)(jit)(torch.jit.trace)(torch.jit.script)](https://blog.csdn.net/hxxjxw/article/details/120835884)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"] [ .reference_list ]

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值