【mxnet转onnx报错】Unrecognized attribute: spatial for operator BatchNormalization

最新推荐文章于 2023-10-10 20:24:40 发布

原创

最新推荐文章于 2023-10-10 20:24:40 发布 · 1.3k 阅读

标签

#mxnet #python #深度学习

收录于

在尝试将MXNet 1.7.0模型导出为ONNX 1.7.0格式时遇到错误，报错信息提及未识别BatchNormalization操作符的'spatial'属性。解决方法是降级ONNX到1.3.0版本，卸载并重新安装后，模型成功导出。

问题描述

mxnet版本：1.7.0 (latest)
onnx版本：1.7.0 (latest)
按照Exporting to ONNX format教程中的方法导出mxnet模型，报错情况如下：

>>> python mx2onnx.py
INFO:root:Converting json and weight file to sym and params
[10:45:36] src/nnvm/legacy_json_util.cc:209: Loading symbol saved by previous version v1.2.0. Attempting to upgrade...
[10:45:36] src/nnvm/legacy_json_util.cc:217: Symbol successfully upgraded!
Traceback (most recent call last