torch.onnx.export转模型遇到的错误

最新推荐文章于 2025-02-14 17:45:57 发布
原创 最新推荐文章于 2025-02-14 17:45:57 发布 · 2.4k 阅读 · 4 ·
CC 4.0 BY-SA版权
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。

本文介绍了解决在使用PyTorch构建模型时遇到的一个常见错误：RuntimeError: Cannot insert a Tensor that requires grad as a constant。该错误通常发生在尝试将需要梯度计算的张量作为常量插入模型中时。解决方法是使用PyTorch提供的nn.ModuleList替代Python原生的list([])，以确保所有需要计算梯度的张量都能正确地被跟踪。

# 遇到的错误

RuntimeError: Cannot insert a Tensor that requires grad as a constant. Consider making it a parameter or input, or detaching the gradient

# 解决办法

把模型构建过程中的list([])换成pytorch原生的nn.ModuleList()。

hukongtao1994
关注
pytorch转onnx遇到的问题汇总
juluwangriyue的博客
06-11 3844
1、关于torch导出onnx时候无法导出 upsample_bilinear2d的问题： 有人说直接将bilinear换成nearest，但是模型效果自然打折扣 完美的解决方案如下 torch.onnx.export(model, dummy_tensor, output, export_params=True,opset_version=11) 这里只需要在后面多加一个 opset_version = 11 就可以解决 但是前提是你的pytorch版本需要大于1.3.0 2、ONNX export f
torch.onnx.export()函数详解
qq_33287871的博客
07-10 7730
torch.onnx.export(model, args, f, export_params=True, verbose=False, training=False, input_names=None, output_names=None, aten=False, export_raw_ir=False, operator_export_type=None, opset_version=None, _retain_param_name=True, do_constant_folding=False, ex
Torch 转 ONNX遇到UnsupportedOperatorError: Exporting the operator ::resolve_conj to ONNX opset version
Y_飞鱼
07-05 2484
【代码】Torch 转 ONNX遇到UnsupportedOperatorError: Exporting the operator ::resolve_conj to ONNX opset version。
PyTorc添加网络图结构add_graph报错：RuntimeError: Cannot insert a Tensor that requires grad as a constant.
江前云后的专栏
02-01 7523
错误说明 在使用PyTorch自带的TensorBoard的add_graph方法将网络图结构添加到监测信息中时，遇到如下报错： RuntimeError: Cannot insert a Tensor that requires grad as a constant. Consider making it a parameter or input, or detaching the gradie...
pytorch导出onnx格式时报错，RuntimeError: Failed to export an ONNX attribute, since it's not constant, please
mjiansun的专栏
10-15 3622
运行程序，Pytorch版YOLOv3，链接地址： https://github.com/eriklindernoren/PyTorch-YOLOv3 Pytorch版本：1.1.0 onnx转换程序： import torch import torchvision from models import * import argparse def main(): par...
torch导出ONNX模型报错：OnnxExporterError: Module onnx is not installed
最新发布
hongweigg的专栏
02-14 3124
使用torch 导出模型为onnx文件时报错：torch.onnx.OnnxExporterError: Module onnx is not installed!
PyTorch的torch.onnx.export函数导出包含多个输出的模型
liugan528的博客
01-01 1289
PyTorch的torch.onnx.export函数导出包含多个输出的模型
torch.onnx.export详细介绍
一直在水些技术小文
06-21 1万+
函数原型参数介绍mode (torch.nn.Module, torch.jit.ScriptModule or torch.jit.ScriptFunction)args (tuple or torch.Tensor)fexport_params (bool, default True)verbose (bool, default False)training (enum, default TrainingMode.EVAL)input_names (list of str, default empty
[pytorch] torch.onnx.export 使用及自定义层构建
头狼博客
07-14 1979
构建pytorch转onnx的自定义层
torch.onnx.export():将pytorch模型转换为.onnx模型
热门推荐
QFJIZHI的博客
04-01 4万+
torch.onnx.export(model, args, f, export_params=True, verbose=False, training=False, input_names=None, output_names=None, aten=False, export_raw_ir=False, operator_export_type=None, opset_version=Non...
[机器学习]Pytorch导出onnx报错“RuntimeError: Cannot insert a Tensor that requires grad as a constant...“
Jamesika 的客栈
08-08 2506
pytorch导出onnx报错"RuntimeError: Cannot insert a Tensor that requires grad as a constant. Consider making it a parameter or input, or detaching the gradient"的一种解决方法
torch转onnx错误 exit code -1073741819
weixin_46412999的博客
06-27 419
torch转onnx错误 exit code -1073741819 Process finished with exit code -1073741819 (0xC0000005) 原因: self.p5_td_w1 = nn.Parameter(torch.ones(1)) 返回
关于 ONNX export failed: Couldn't export operator aten::upsample_bilinear2d 解决方案
info_black_hole的博客
04-16 3524
关于torch导出onnx时候无法导出 upsample_bilinear2d的问题： 有人说直接将bilinear换成nearest，但是模型效果自然打折扣 完美的解决方案如下 torch.onnx.export(model, dummy_tensor, output, export_params=True,opset_version=11) 这里只需要在后面多加一个 opset_versio...
关于 ONNX export failed: Couldn‘t export operator aten::upsample_bilinear2d 解决方案
邹小驴
09-10 903
关于torch导出onnx时候无法导出 upsample_bilinear2d的问题： 有人说直接将bilinear换成nearest，但是模型效果自然打折扣完美的解决方案如下 torch.onnx.export(model, dummy_tensor, output, export_params=True,opset_version=11) 这里只需要在后面多加一个 opset_version = 11 就可以解决 但是前提是你的pytorch版本需要大于1.3.0 或者 http..
onnxsim时候一个跟ir_version相关的错误
h12321hzhy的专栏
07-26 3573
github上查了各种信息也没有很靠谱的解决方案，经过无心插柳的查找，发现原来是onnx把model序列化成字串的时候，用了protobuf，这个版本高的话，会让序列化字串出错，Onnx导不进来，把该pip包降到了。无语死了，想优化一个ChatGLM2，结果导出的onnx，用onnxsim简化不了，一直提示。之后，成功载入并且sim成功，记录一下，赠给有缘人。
torch自定义算子转onnx模型报错
qq_42001765的博客
11-24 3455
【代码】torch自定义算子转onnx模型报错。
导出onnx模型(2)：torch.onnx.export 单输入
weixin_43470772的博客
05-16 371
将模型的输入直接传输到torch.onnx.export()函数中即可。
导出onnx模型(1)：torch.onnx.export 多输入
weixin_43470772的博客
05-06 2158
将模型的多个输入构造为dict，输入到torch.onnx.export（）函数中，实现多输入模型导出onnx。
[pytorch][转载]torch.onnx.export():将pytorch模型转换为.onnx模型
FL1623863129的博客
11-22 3967
torch.onnx.export(model, args, f, export_params=True, verbose=False, training=False, input_names=None, output_names=None, aten=False, export_raw_ir=False, operator_export_type=None, opset_version=None, _retain_param_name=True, do_constant_folding=False, e.
torch.onnx._export
09-14
torch.onnx.export是PyTorch中的一个函数，用于将模型导出为ONNX格式的文件。它接受多个参数，包括模型、输入数据、保存路径等。 首先，你需要导入torch和torchvision库。 然后，你可以使用torch.onnx.export函数来...
评论 1
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值