search

rockchip-linux / rknn-toolkit

BSD 3-Clause "New" or "Revised" License
825 stars 175 forks source link

RKNN - LSTM not support for non-zero h_state, c_state #385

Open andrewhuman opened 1 year ago

andrewhuman commented 1 year ago

\root@db5694cd7562:/MASR/rknn_convert# python rknn_conversion_with_hidden.py

../lstm_1_1_67_80_ctc_161024_with_hidden.pt **** W Input channel 1 less then 3, please update NPU Driver to v1.3.0 or later. W Input channel 1 less then 3, please update NPU Driver to v1.3.0 or later. W Input channel 1 less then 3, please update NPU Driver to v1.3.0 or later. E RKNN - LSTM not support for non-zero h_state, c_state E Catch exception when loading pytorch model: ../lstm_1_1_67_80_ctc_161024_with_hidden.pt! E Traceback (most recent call last): E File "rknn/base/RKNNlib/converter/convert_pytorch_new.py", line 5127, in rknn.base.RKNNlib.converter.convert_pytorch_new.convert_pytorch.load E File "rknn/base/RKNNlib/converter/convert_pytorch_new.py", line 5301, in rknn.base.RKNNlib.converter.convert_pytorch_new.convert_pytorch.parse_nets E File "rknn/base/RKNNlib/converter/convert_pytorch_new.py", line 4915, in rknn.base.RKNNlib.converter.convert_pytorch_new.PyTorchOpConverter.convert_operators E File "rknn/base/RKNNlib/converter/convert_pytorch_new.py", line 1298, in rknn.base.RKNNlib.converter.convert_pytorch_new.PyTorchOpConverter.lstm E File "rknn/api/rknn_log.py", line 323, in rknn.api.rknn_log.RKNNLog.e E ValueError: RKNN - LSTM not support for non-zero h_state, c_state E Please feedback the detailed log file to the RKNN Toolkit development team. E You can also check github issues: https://github.com/rockchip-linux/rknn-toolkit/issues

Load Pytorch model failed!

how do i fix it?