Toybrick

标题: onnx转rknn，LSTM层报错 [打印本页]

作者: cfan 时间: 2021-5-13 15:25
标题: onnx转rknn，LSTM层报错
I Try match LSTM_105

ut0
W Not match tensor LSTM_105

ut0
E Try match LSTM_105

ut0 failed, catch exception!
W ----------------Warning(2)----------------
E Catch exception when loading onnx model: lstm_model.onnx!
E Traceback (most recent call last):
E File "rknn/base/RKNNlib/converter/convert_onnx.py", line 826, in rknn.base.RKNNlib.converter.convert_onnx.convert_onnx.match_paragraph_and_param
E File "rknn/base/RKNNlib/converter/convert_onnx.py", line 728, in rknn.base.RKNNlib.converter.convert_onnx.convert_onnx._onnx_push_ready_tensor
E TypeError: 'NoneType' object is not iterable
E During handling of the above exception, another exception occurred:
E Traceback (most recent call last):
E File "rknn/api/rknn_base.py", line 264, in rknn.api.rknn_base.RKNNBase.load_onnx
E File "rknn/base/RKNNlib/RK_nn.py", line 135, in rknn.base.RKNNlib.RK_nn.RKnn.load_onnx
E File "rknn/base/RKNNlib/app/importer/import_onnx.py", line 121, in rknn.base.RKNNlib.app.importer.import_onnx.Importonnx.run
E File "rknn/base/RKNNlib/converter/convert_onnx.py", line 832, in rknn.base.RKNNlib.converter.convert_onnx.convert_onnx.match_paragraph_and_param
E File "rknn/api/rknn_log.py", line 312, in rknn.api.rknn_log.RKNNLog.e
E ValueError: Try match LSTM_105ut0 failed, catch exception!

onnx转rknn报这个错，请问该怎么解决?

作者: jefferyzhang 时间: 2021-5-13 15:30
Try match LSTM_105
这个就是典型的OP不支持而已。。。。

作者: cfan 时间: 2021-5-13 15:33

jefferyzhang 发表于 2021-5-13 15:30
Try match LSTM_105
这个就是典型的OP不支持而已。。。。

哦哦，但是我看OP支持列表ONNX是支持LSTM的，请问是有什么限制吗？

作者: jefferyzhang 时间: 2021-5-13 15:39

cfan 发表于 2021-5-13 15:33
哦哦，但是我看OP支持列表ONNX是支持LSTM的，请问是有什么限制吗？

建议你进rknn官方qq群里问下他们。

欢迎光临 Toybrick (https://t.rock-chips.com/)