概述
本文主要解决Mxnet 转onnx,以及onnx模型caffe2推理中遇到的两个可能的报错。
(1)Don’t know how to map unexprected argument spatial(from operator SpatialBN)
(2)onnx.onnx_cpp2py_export.checker.ValidationError: Unrecognized attribute:spatial for operator BatchNormalization
问题:
(1)mxnet<=1.4 版本,对于batchnorm 可以转换成功,但是在使用caffe2.python.onnx.backend.prepare() 加载onnx模型时出现 报错:
Don’t know how to map unexprected argument spatial(from operator SpatialBN)
再使用netron 查看onnx模型,选择batchnormlization 模块,发现Attributes属性多了spatial:0项,从而导致再caffe2 推理时不支持/不识别spatial参数。
填坑尝试:
一顿查找后发现这是mxnet的bug,官方给出的说法是