Toybrick

标题: load onnx 模型出错 [打印本页]

作者: wdw    时间: 2019-3-11 10:16
标题: load onnx 模型出错
pytorch模型转onnx,加载onnx模型时报错如下:
E Catch exception when loading onnx model: ./mnv2_deconv2.onnx!
T Traceback (most recent call last):
T   File "rknn/api/rknn_base.py", line 321, in rknn.api.rknn_base.RKNNBase.load_onnx
T   File "rknn/base/rknnlib/converter/convert_onnx.py", line 555, in rknn.base.rknnlib.converter.convert_onnx.convert_onnx.match_paragraph_and_param
T   File "rknn/base/rknnlib/converter/convert_onnx.py", line 355, in rknn.base.rknnlib.converter.convert_onnx.convert_onnx._onnx_build_acu_layer
T   File "rknn/base/rknnlib/converter/convert_onnx.py", line 322, in rknn.base.rknnlib.converter.convert_onnx.convert_onnx._onnx_acu_param_assign
T   File "rknn/base/rknnlib/converter/convert_onnx.py", line 302, in rknn.base.rknnlib.converter.convert_onnx.convert_onnx._onnx_parse_param
T   File "rknn/base/rknnlib/converter/paragraph.py", line 21, in rknn.base.rknnlib.converter.paragraph.lambda4
T TypeError: 'int' object is not iterable
load model failed




作者: elooon    时间: 2019-3-11 16:47
u can set verbose=True to get more info
作者: wdw    时间: 2019-3-11 18:12
elooon 发表于 2019-3-11 16:47
u can set verbose=True to get more info

W verbose file path is invalid, debug info will not dump to file.
I Current ONNX Model use ir_version 3 opset_version 9
D import clients finished
I build output layer output_473
I Try match Conv op#154
I Match [['473']] [['Conv_1']] to [['convolution_1']]
I Try match Relu op#153
I Match [['472']] [['Relu_1']] to [['relu_1']]
I Try match BatchNormalization op#152
I Match [['471']] [['BatchNormalization']] to [['batchnormalize_1']]
I Try match ConvTranspose op#151
I Match [['470']] [['ConvTranspose_1']] to [['deconvolution_1']]
E Catch exception when loading onnx model: ./model.onnx!
T Traceback (most recent call last):
T   File "rknn/api/rknn_base.py", line 321, in rknn.api.rknn_base.RKNNBase.load_onnx
T   File "rknn/base/rknnlib/converter/convert_onnx.py", line 555, in rknn.base.rknnlib.converter.convert_onnx.convert_onnx.match_paragraph_and_param
T   File "rknn/base/rknnlib/converter/convert_onnx.py", line 355, in rknn.base.rknnlib.converter.convert_onnx.convert_onnx._onnx_build_acu_layer
T   File "rknn/base/rknnlib/converter/convert_onnx.py", line 322, in rknn.base.rknnlib.converter.convert_onnx.convert_onnx._onnx_acu_param_assign
T   File "rknn/base/rknnlib/converter/convert_onnx.py", line 302, in rknn.base.rknnlib.converter.convert_onnx.convert_onnx._onnx_parse_param
T   File "rknn/base/rknnlib/converter/paragraph.py", line 21, in rknn.base.rknnlib.converter.paragraph.lambda4
T TypeError: 'int' object is not iterable
load model failed





欢迎光临 Toybrick (https://t.rock-chips.com/) Powered by Discuz! X3.3