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

发布时间 2023-03-22 21:11:22作者: singyoutosleep

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这一行注释掉