mxnet模型转onnx报错onnx.onnx_cpp2py_export.checker.ValidationError: Unrecognized attribute: spatial for operator BatchNormalization

onnx 版本

onnx 1.6.0
onnx-simplifier 0.2.2
onnxruntime 1.1.0
onnxruntime-gpu 1.1.0

mxnet 版本

mxnet-cu101       1.7.0

原因 mxnet bug

https://github.com/apache/mxnet/issues/16439#issuecomment-543860499

解决方法

vim ～/anaconda3/envs/face/lib/python3.6/site-packages/mxnet/contrib/onnx/mx2onnx/_op_translations.py

找到spatial=0这一行注释掉

 

posted @   singyoutosleep  阅读(275)  评论(0)  编辑  收藏  举报
相关博文：
·  InvalidVersionSpecError: Invalid version spec: =2.7
·  ImportError: /lib64/libstdc++.so.6: version `GLIBCXX_3.4.20' not found (required by xxx)
·  onnx转ncnn报错Shape not supported yet! Gather not supported yet # axis=0 Unsupported unsqueeze axes ! U
·  SAM(segment-anything导出onnx模型报错unsupported onnx opset version：17)
·  torch 导出onnx RuntimeError: Exporting the operator _convolution_mode to ONNX opset version 12 is not supported
阅读排行：
· 25岁的心里话
· 闲置电脑爆改个人服务器（超详细） #公网映射 #Vmware虚拟网络编辑器
· 零经验选手，Compose 一天开发一款小游戏！
· 通过 API 将Deepseek响应流式内容输出到前端
· AI Agent开发，如何调用三方的API Function，是通过提示词来发起调用的吗
点击右上角即可分享
微信分享提示