onnxruntime错误Expand fails with -1 in the size

最新推荐文章于 2024-06-20 14:52:58 发布
最新推荐文章于 2024-06-20 14:52:58 发布
阅读量1.3k 收藏 1
点赞数
分类专栏： pytorch 文章标签： onnx pytorch
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/u011622208/article/details/102464336
版权

具体报错：
RuntimeError: Method run failed due to: [ONNXRuntimeError] : 1 : GENERAL ERROR : /onnxruntime_src/onnxruntime/core/providers/cpu/math/element_wise_ops.h:324 void onnxruntime::BroadcastIterator::Init(int64_t, int64_t) axis == 1 || axis == largest was false. Attempting to broadcast an axis by a dimension other than 1. -1 by 32
Stacktrace:

github同样的问题
https://github.com/microsoft/onnxruntime/issues/1840

解决方案

  1. 用torch.cat方式
    src_seq = torch.cat([src_seq for _ in range(h)], dim=1)

农夫山泉2号
关注
  • 0
    点赞
  • 踩
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
void onnxruntime::BroadcastIterator::Append(int64_t, int64_t) axis == 1 || axis == largest was false
ltochange的博客
10-15 1810
import torch import torch.nn as nn import os from allennlp.nn import util torch.manual_seed(1) os.environ["CUDA_VISIBLE_DEVICES"] = "1" class Test(nn.Module): def __init__(self): super().__init__() pass def forward(self, mix, of
PyTorch转onnx转TRT中不支持的操作
Jamie的技术博客
06-09 1241
PyTorch训练的模型，在转为ONNX再转为TRT模型时，有一些操作不支持 假设输入是AvgPooling以后的特征 torch.nn.functional.normalize会引入Expand操作 解决方案： F.normalize(features, p=2, dim=1) 替换为features.div(features.norm(p=2, dim=1, keepdim=True)) size()会引入Gather操作 解决方案： onnxsim转化 features.view(feat
onnxruntime.capi.onnxruntime_pybind11_state.InvalidProtobuf: [ONNXRuntimeError] : 7 : INVALID_PROTOB
最新发布
yuchunlei的专栏
06-20 526
OnnxRuntime加载模型报错的问题原因之一： 模型数据的名称不是默认的 model.onnx时，如果只指定目录就会报上面的错误，指定到具体的文件名就正常了。
ReXNet模型pth转onnx,swish不支持问题
风华正茂的博客
10-19 887
ReXNet模型转onnx,swish不支持问题
onnxruntime报错
怡宝2号
10-09 9977
onnxruntime报错： Exception during initialization: /onnxruntime_src/onnxruntime/core/providers/cpu/math/element_wise_ops.h:324 void onnxruntime::BroadcastIterator::Init(int64_t, int64_t) axis == 1 || axi...
Ubuntu20.04安装CUDA、cuDNN、onnxruntime、TensorRT
weixin_42156097的博客
10-14 5048
Ubuntu20.04系统安装CUDA、cuDNN、onnxruntime、TensorRT
使用element-ui table expand展开行实现手风琴效果
10-17
为了实现手风琴效果，我们需要利用Element-UI提供的`toggleRowExpansion`方法以及`row-key`和`expand-row-keys`属性。 1. **toggleRowExpansion方法**： 这个方法用于切换表格行的展开状态。它接收两个参数：`row`...
bootstrap table方法之expandRow-collapseRow展开或关闭当前行数据
08-29
Bootstrap Table 中的 expandRow 和 collapseRow 方法 Bootstrap Table 是一个功能强大且灵活的表格插件，它提供了许多实用的方法来处理表格数据。今天，我们将详细介绍 Bootstrap Table 中的 expandRow 和 ...
ansible-role-ccdc-expand-artifactory-archives
03-17
角色变量剧本范例- hosts: all roles: - ccdc-expand-artifactory-archives vars: artifactory-archives: # Executables used by our software - artifactory_path: ccdc-3rdparty-windows-runtime-exes filename: ...
expand-cell-fullscreen：将Jupyter笔记本电脑单元扩展到整个屏幕
02-04
安装克隆或下载此存储库： git clone https://github.com/scottlittle/expand-cell-fullscreen使用类似命令安装扩展： 如果您不希望Jupyter将文件复制到Jupyter数据路径，则可以使用--symlink选项。 同样，选项--sys...
expand-search-two
03-21
在"expand-search-two"这个主题下，我们可以推测这是一个与扩展搜索框相关的项目，可能涉及到CSS的布局、交互效果以及可能的动画设计。下面将详细探讨与这个主题相关的CSS知识点。 1. **布局**：在网页设计中，布局...
解决 [ONNXRuntimeError] : 1 : FAIL Non-zero status code returned
05-16
pytorch转onnx模型后，对onnx模型进行runtime时提示以下错误，具体细节如下： onnxruntime.capi.onnxruntime_pybind11_state.Fail: [ONNXRuntimeError] : 1 : FAIL : Non-zero status code returned while running FusedConv node. Name:'Conv_461_Add_463_Relu_464' Status Message: CUDNN error executing cudnnSetTensorNdDescriptor(tensor_, dataType, static_cast<int>(rank), dims.data(), strides.data())。 pdf文件中是对该错误的解决思路。
docker nvidia-smi Failed to initialize NVML: Driver/library version mismatch RuntimeError: cuda_call
hknaruto的专栏
06-26 526
x86_64 ubuntu20.04操作系统上制作的docker，在深度20.8上运行测试报错python报错。
onnxruntime-gpu报错问题
jinxianwei1999的博客
11-30 3657
问题来源：使用onnxruntime推理时，总是使用cpu；
ONNXRuntimeError] : 2 : INVALID_ARGUMENT : Failed to load model with error: /onnxruntime_src/onnxrun
binqiang2wang
09-02 3354
ONNXRuntimeError] : 2 : INVALID_ARGUMENT : Failed to load model with error: /onnxruntime_src/onnxruntime/core/graph/model_load_utils.h:47 void 运行spikingjelly遇到的错误，运行时错误： 具体原因没有源码分析（牵扯到.h了，已经不是python的范围） 但解决方案是修改onnx的版本： pip uninstall onnx pip install onnx
上采样pytorch转onnx之onnx runtime error，以及tensorrt解析onnx的错误（Floor int32）
随便记记
04-26 2676
pytorch->onnx错误 出现场景 pytorch（v1.1.0）训练了一个Unet网络，其中用到了上采样层，最开始直接用torch.onnx.export转onnx(Opset=9)时出现错误： RuntimeError: Failed to export an ONNX attribute, since it’s not constant, please… 原因 就是pytorch...
MXNet转onnx问题点记录
zhaokai5的博客
09-08 1008
MXNet转onnx问题点记录MXnet转onnx时碰到的问题记录主要问题总结 MXnet转onnx时碰到的问题记录 最近将mxnet转onnx时碰到很多问题，在查找过程中发现解决方法，特记录下；转换过程主要参考如下链接中的解决方法 mxnet模型转onnx模型 作者liguiyuan112 Insightface中ArcFace MxNet2ONNX踩坑 MXNet Symbol Batch Normalization fix_gamma=True转ONNX方法 主要问题 转换时BN层出现错误 on
onnxruntime-gpu飘红报错怎么解决？
j_starry的博客
04-07 728
这几天在搭comfyui环境的时候遇到一个棘手的问题，使用insightface的时候，GPU报错，只能用CPU，虽然能够运行，但也降低了运行速度。原本我是1.17.0的版本，cuda的话应该比较高，尝试了1.16多的版本也不行，如果有这个困扰的小伙伴，可以用这个方式，去找下比较搭配的版本。报错有少许不同，就类似这种，不用查也知道是onnxruntime-gpu版本的问题，但一直不知道该是哪个版本，今天看到个解决方法，终于有用了。这样安装的是1.17.1的版本。祝大家成功赶走bug！
使用YOLOv7进行Transformer检测模型的TensorRT部署
jinfagang1211的博客
04-23 9506
最近有大佬开源了YOLOv7, 大概看了一眼， 支持DETR等模型的ONNX导出，并且可以进行tensorrt推理。但是原作者并没有开源对应的权重，也没有开源onnx推理的脚本。本文就依托YOLOv7这个项目，将DETR导出到onnx,并教大家如何使用这个onnx进行tensorrt的部署。 首先clone一下原项目： https://github.com/jinfagang/yolov7 DETR 权重推理验证 原来的项目其实也支持AnchorDETR,这个我没有测试，但是
el-table expand-row-keys 失效
12-12
根据提供的引用内容，el-table expand-row-keys 失效可能是由于以下原因导致的： 1. expand-row-keys属性的值不正确或为空。 2. 数据源中的keyField字段与expand-row-keys属性的值不匹配。 3. 在设置expand-row-keys属性之前，数据源还没有加载完成。 4. 在设置expand-row-keys属性之前，el-table组件还没有渲染完成。 为了解决这个问题，可以尝试以下方法： 1. 确保expand-row-keys属性的值正确，并且不为空。 2. 确保数据源中的keyField字段与expand-row-keys属性的值匹配。 3. 确保在设置expand-row-keys属性之前，数据源已经加载完成。 4. 确保在设置expand-row-keys属性之前，el-table组件已经渲染完成。 以下是一个设置expand-row-keys属性的例子： ```html <el-table :data="tableData" :expand-row-keys="expandRowKeys"> <el-table-column type="expand" width="0"> <template slot-scope="props"> <div>这里是扩展内容</div> </template> </el-table-column> <el-table-column prop="name" label="姓名"></el-table-column> <el-table-column prop="age" label="年龄"></el-table-column> </el-table> ``` ```javascript export default { data() { return { tableData: [ { name: '张三', age: 18 }, { name: '李四', age: 20 }, { name: '王五', age: 22 } ], expandRowKeys: [0, 1] // 设置需要展开的行的key值 } } } ```
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值