第1个回答 2022-07-26
本文记录基于mxnet训练得到的resnet50模型在转为onnx格式部署时发生的两个错误及解决办法
os: ubuntu 16.04
Mxnet : 1.6.0
onnx: 1.6.0
cuda: 10.2
cudnn: 8.0
resnet50中使用的batchnorm层在转换至onnx时报错不支持属性spatial
mxnet源码mxnet/contrib/onnx/mx2onnx/_op_translation.py 359行 注释掉spatial参数
通过netron.app 查看网络结构, 看到第一个batchnorm层(bn_data)的参数bn_data_gamma异常:
在onnx模型中手动修改bn_data层的gamma参数为1.0