pytorch保存模型参数信息与模型结构信息

最新推荐文章于 2023-06-08 09:50:35 发布
最新推荐文章于 2023-06-08 09:50:35 发布
阅读量3.4k 收藏 1
点赞数
分类专栏： 人工智能 文章标签： python pytorch
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/Grandret/article/details/109812699
版权
本文介绍了PyTorch中使用torch.jit.save和torch.jit.load来保存和加载模型的优势，如避免引入原始模型脚本，实现跨平台使用。通过示例代码展示如何保存和加载 TorchScript 模型，同时提醒注意保存的是TorchScript格式，并提及可能遇到的错误。此外，还提到torch.jit.save能保存不依赖第三方库的函数，但可能会遇到找不到特定模块的AttributeError。
摘要由CSDN通过智能技术生成

阅读官网文档发现pytorch除了使用torch.save与torch.load保存加载数据/模型外还可以使用torch.jit.save(),torch.jit.load()。
后者的好处在于

  1. 其他py脚本在使用模型时不需要将原模型所在py文档也引入
  2. 可以跨平台使用

官网代码

scripted_module = torch.jit.script(MyModule())
torch.jit.save(scripted_module, 'mymodule.pt')
torch.jit.load('mymodule.pt')

以上代码即可，注意torch.jit.save保存的时torchscript模型，所以要先进行转换。否则报错如下：

torch.nn.modules.module.ModuleAttributeError: 'Model
最低0.47元/天 解锁文章
阅读空气
关注
专栏目录
pytorch----模型参数的保存与加载
qq_38055396的博客
04-06 1368
网络模型的保存与加载 1.单纯保存网络模型参数，一条语句即可 torch.save(model.state_dict(), path) # 其中path=’./model.pth’ , path=’./model.tar’, path=’./model.pkl’ # 保存参数的文件一定要有后缀扩展名。 model.load_state_dict(torch.load(path)) 2.还想保存训练采用的优化器、epoch信息等 state = { 'model': model.state_dict
torch.save()
RUN
09-27 5万+
pytorch保存模型等相关参数，利用torch.save()，以及读取保存之后的文件 本文分为两部分，第一部分讲如何保存模型参数，优化器参数等等，第二部分则讲如何读取。 假设网络为model = Net(), optimizer = optim.Adam(model.parameters(), lr=args.lr), 假设在某个epoch，我们要保存模型参数，优化器参数以及epoch 一、save 先建立一个字典，保存三个参数： state = {‘net’:model.state_dict(),
.torch.save与torch.jit.save
jianminfly的博客
06-21 4595
1.torch.jit.savetorch.jit.save用来保存编译后的模型，支持跨平台，要注意模型中只能使用pytorch的函数。jit.save支持保存script类型和trace类型的模型，其中script为全量模型，trace为仅保存运行过的路径2.torch.savetorch.save有三种使用场景，参考：https://stackoverflow.com/questions/42703500/best-way-to-save-a-trained-model-in-pytorch1.保存模型
torch.save
weixin_36670529的博客
05-04 3097
torch.save(obj, f, pickle_module=<module 'pickle' from '/opt/conda/lib/python3.6/pickle.py'>, pickle_protocol=2, _use_new_zipfile_serialization=False)[source] 讲对象保存为磁盘文件。 See also: Recommended approach for saving a model 参数： obj – 保存对象 f –
torch保存和加载 模型、参数
最新发布
zoe9698的博客
06-08 518
缺点：这种方式保存的模型只能在相同的环境中使用，因为它依赖于模型定义的代码。如果您在不同的环境中使用这种方式保存的模型，可能会出现错误。（相同的环境指的是运行代码的环境，包括操作系统、Python 版本、PyTorch 版本以及其他依赖库的版本都相同。来保存和加载模型的参数。这样，您只需要在新环境中定义相同的模型结构，然后使用。优点：想要在不同的环境中使用保存的模型，建议使用。保存整个模型，包括模型结构+参数。函数加载保存的参数即可。
基于pytorch的保存和加载模型参数的方法
09-18
#### 保存模型参数： ```python # 定义一个模型net net = MyModel() # 训练模型... # 保存模型参数 torch.save(net.state_dict(), 'model_params.pth') ``` 在这里，`net.state_dict()`返回一个字典，包含了模型中...
pytorch保存模型pth_pytorch中保存的模型文件.pth深入解析
weixin_30091863的博客
12-24 1843
前言：前面有专门的讲解关于如何深入查询模型的参数信息，可以参考这篇文章：本次来解析一下我们通常保存的模型文件 .pth 文件到底内部是什么？一、.pth 文件详解在pytorch进行模型保存的时候，一般有两种保存方式，一种是保存整个模型，另一种是只保存模型的参数。torch.save(model.state_dict(), “my_model.pth”) # 只保存模型的参数torch.save(...
python保存模型与参数_Pytorch - 模型和参数的保存与恢复
weixin_39646706的博客
03-02 1305
模型训练后，需要保存到文件，以供测试和部署；或，继续之前的训练状态.1. Best Practices主要有两种模型序列化保存和加载恢复的方法.1.1 方法 M1 - 推荐只保存和加载恢复模型参数(model parameters)：import torch# 保存torch.save(the_model.state_dict(), PATH)# 恢复the_model = TheModelCla...
pytorch保存模型参数
05-24
在 PyTorch 中，可以使用 `torch.save()` 方法将模型的参数保存到磁盘上。具体的代码示例如下： ...需要注意的是，加载模型参数时，模型的结构必须与保存模型参数时的结构一致，否则会导致加载失败。
pytorch模型保存
Bolly_He的博客
05-27 129
目录 只保存模型训练参数 保存完整的模型 只保存模型训练参数 # 保存 torch.save(model.state_dict(), '\parameter.pkl') # 加载 model = ModelClass(...) model.load_state_dict(torch.load('\parameter.pkl')) 保存完整的模型 # 保存 torch.save(model, '\model.pkl') # 加载 model = torch.load('\model.pk.
pytorch保存模型
Rou_l的博客
03-22 364
pytorch 状态字典:state_dict pytorch学习笔记（五）：保存和加载模型 pytorch保存模型官方文档
Python基本语法——模型的保存torch.save()
AIChemist
03-07 4218
转载 pytorch保存模型等相关参数，利用torch.save()，以及读取保存之后的文件
torch.save() torch.load()
t20134297的博客
11-06 2394
在使用pytorch进行深度学习训练的时候，经常会有些内容需要保存下来，保存到硬盘张，不管什么时候我们都可以读取到，那么这个时候，使用torch.save()就可以将内容存储器来，使用torch.load()就可以将存取的内容读取出来。 1.torch.save()使用方法 torch.save()使用的时候需要往里面放进去两个参数，分别是要存储的内容和存储的地址。下面是个例子： import torch statemement = {'name':'dalishi','age':19} torc
Pytorch：模型的保存与加载 torch.save()、torch.load()、torch.nn.Module.load_state_dict()
热门推荐
weixin_40522801的博客
06-05 21万+
Pytorch 保存和加载模型后缀：.pt 和.pth 1 torch.save()[source] 保存一个序列化（serialized）的目标到磁盘。函数使用了Python的pickle程序用于序列化。模型（models），张量（tensors）和文件夹（dictionaries）都是可以用这个函数保存的目标类型。 torch.save(obj, f, pickle_module=<module '...'>, pickle_protocol=2) 参数 描述 o..
Pytorch—模型的保存和加载torch.save()、torch.load()
weixin_44296865的博客
11-20 1746
pytorch中torch.save()、torch.load()用于模型的存储和加载
torch.save()用法
weixin_38410551的博客
09-27 4万+
torch.save()用法：保存模型参数 参考链接： https://www.cnblogs.com/qinduanyinghua/p/9311410.html
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值