关于 ONNX export failed: Couldn‘t export operator aten::upsample_bilinear2d 解决方案

最新推荐文章于 2024-02-21 18:24:51 发布
最新推荐文章于 2024-02-21 18:24:51 发布
阅读量3.7k 收藏 2
点赞数
分类专栏： 学习笔记 文章标签： TensorRT
原文链接：https://blog.csdn.net/info_black_hole/article/details/105556670
版权

关于torch导出onnx时候无法导出 upsample_bilinear2d的问题：
有人说直接将bilinear换成nearest，但是模型效果自然打折扣
完美的解决方案如下

torch.onnx.export(model, dummy_tensor, output, export_params=True,opset_version=11)
1
这里只需要在后面多加一个 opset_version = 11 就可以解决
但是前提是你的pytorch版本需要大于1.3.0

Decoder-Wang
关注
  • 0
    点赞
  • 踩
  • 2
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
专栏目录
Pytorch——初探onnx（1）解决upsample_bilinear2d转换问题
weixin_45709671的博客
07-07 4052
问题 将pytorch的pth文件读取后转换为ONNX格式，尝试使用下述代码将model导出为onnx格式 onnx.export(model, input_image, 'model.onnx') 出现了如下错误 RuntimeError: ONNX export failed: Couldn‘t export operator aten::upsample_bilinear2d 环境 目前测试发现pytorch1.6及以下均存在此问题 复现 下述代码中使用的mode为bilinear，align_co
RuntimeError: ONNX export failed: Couldn't export operator aten::avg_pool2d
孙荣达的博客
05-24 6519
RuntimeError: ONNX export failed: Couldn’t export operator aten::avg_pool2d 用ONNX做模型转换从pytorch转换成ONNX做算法移植，resNet50，32都没问题，但是到resNet18，ONNX模型无法导出报错。 看了一下问题，avg_pool2d层的问题复上源码 首先，如果你是使用了avg_pool2d，那很好办...
ONNX export failed: Couldn‘t export operator aten::upsample_bilinear2d
so
08-07 381
升级pytroch >1.3 opset_version=10 torch.onnx.export(model, dummy_tensor, output, export_params=True,opset_version=11)
Couldn‘t export operator aten::upsample_bilinear2d
baidu_34172099的博客
03-30 1194
这是在pytorch在向onnx转移模型时候遇到的问题，虽然onnx并不是完全支持pytorch所有层，但是上采样这个还是可以的，导致这个的原因是upsample_blinear2d丢失导致，所以在envs下的pytorch的onnx文件夹下找到 symbolic_opset9.py，把里面720行左右相似的语句修改为如下（建议修改之前先复制一份文件备份）： upsample_bilinear1d...
关于 ONNX export failed: Couldn't export operator aten::upsample_bilinear2d 解决方案
info_black_hole的博客
04-16 3232
关于torch导出onnx时候无法导出 upsample_bilinear2d的问题： 有人说直接将bilinear换成nearest，但是模型效果自然打折扣 完美的解决方案如下 torch.onnx.export(model, dummy_tensor, output, export_params=True,opset_version=11) 这里只需要在后面多加一个 opset_versio...
json error: Use of overloaded operator [] is ambiguous错误的解决方法
08-26
今天小编就为大家分享一篇关于json error: Use of overloaded operator [] is ambiguous错误的解决方法，小编觉得内容挺不错的，现在分享给大家，具有很好的参考价值，需要的朋友一起跟随小编来看看吧
shell脚本报错："[: =: unary operator expected"解决办法
09-15
主要介绍了shell脚本报错："[: =: unary operator expected"解决办法的相关资料,需要的朋友可以参考下
sklearn-onnx：将scikit学习模型和管道转换为ONNX
02-05
Linux 视窗介绍sklearn-onnx将模型转换为 。 一旦采用ONNX格式，您就可以使用类的工具进行高性能评分。文献资料包括教程在内的完整文档可在。 您也可以在找到答案或提交新的答案。安装您可以从安装： pip install ...
rxjs-last-operator:用StackBlitz创建:high_voltage:
03-13
rxjs-last-operator
shell脚本报错：”[: =: unary operator expected”解决办法
01-20
shell脚本报错：”[: =: unary operator expected”解决办法  在匹配字符串相等时，我用了类似这样的语句： if [ $STATUS == OK ]; then echo OK fi  在运行时出现了 [: =: unary operator expected 的错误，就...
RuntimeError: ONNX export failed: Couldn‘t export operator aten::upsample_bilinear2d
怡宝2号
09-12 4227
pytorch转onnx时出现： ‘RuntimeError: ONNX export failed: Couldn’t export operator aten::upsample_bilinear2d’ 解决方案 暂时无
RuntimeError: ONNX export failed: Couldn‘t export operator aten::to
binqiang2wang
05-19 867
torch.onnx.export(model, input, onnx_path, verbose=True, input_names=input_names, output_names=output_names) #add:opset_version=11 torch.onnx.export(model, input, onnx_path, verbose=True, input_names=input_names, output_names=output_names, opset_vers...
torch.onnx.errors.UnsupportedOperatorError: Exporting the operator ‘aten::unflatten‘ to ONNX opset
最新发布
Texh1201的博客
02-21 1321
这意味着在当前的 ONNX opset 版本（版本 11）中，不支持将 “aten::unflatten” 运算导出为 ONNX 格式。：如果 “aten::unflatten” 运算在你的模型中被广泛使用，并且目前无法导出为 ONNX 格式，那么你可能需要考虑修改模型结构，以使用 ONNX 支持的运算来替代 “aten::unflatten”。: 你可以尝试使用较低版本的 ONNX opset（如版本 10），因为较低版本的 opset 可能支持更少的运算，但也可能会限制模型的某些功能。
Torch 转 ONNX遇到UnsupportedOperatorError: Exporting the operator ::resolve_conj to ONNX opset version
Y_飞鱼
07-05 1447
【代码】Torch 转 ONNX遇到UnsupportedOperatorError: Exporting the operator ::resolve_conj to ONNX opset version。
pytorch转onnx遇到的问题汇总
juluwangriyue的博客
06-11 3157
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
ONNX export failed on ATen operator grid_sampler because torch.onnx.symbolic.grid_sampler does not e
怡宝2号
09-13 2466
pytorch转onnx时，pytorch中用了F.grid_sample，报错： ‘ONNX export failed on ATen operator grid_sampler because torch.onnx.symbolic.grid_sampler does not exist’ 解决方案 暂时无
Exporting the operator hardsigmoid to ONNX opset version 11 is not supported
jacke121的专栏
06-19 2785
Exporting the operator hardsigmoid to ONNX opset version 11 is not supported
转换为onnx模型错误汇总
06-10
4. ValueError: ONNX export failed: Couldn't export operator aten::upsample_bilinear2d 解决方法：将上采样函数改为 interpolate 函数。 5. RuntimeError: ONNX export failed: Couldn't export operator aten...

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

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

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值