Toybrick

load onnx 模型出错

wdw

新手上路

积分
13
发表于 2019-3-11 10:16:52    查看: 5830|回复: 2 | [复制链接]    打印 | 显示全部楼层
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

注册会员

积分
139
发表于 2019-3-11 16:47:10 | 显示全部楼层
u can set verbose=True to get more info
回复

使用道具 举报

wdw

新手上路

积分
13
 楼主| 发表于 2019-3-11 18:12:18 | 显示全部楼层
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
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表