lstm pytorch->onnx->rknn(rknn1.6)load onnx fail!!!

[ZHR1997]

pytorch中lstm的代码如下： self.rnn = nn.LSTM(nIn, nHidden, num_layers=1, bidirectional=True) 能够成功转换到onnx，但是在onnx转到rknn的时候fail，代码如下 报错如下： 通过Netron查看onnx该层的结构图如下：

我猜想会不会是sequence_lens参数没有导致的？如果是的话，要如何对代码进行修改呢？如果不是的话，又是什么原因导致这个问题的出现，要如何去修改？

[Ajie16]

你好，请问最后时怎么解决的呢？

[ZHR1997]

你好，请问最后时怎么解决的呢？

我用TensorFlow 去重新训练了.......

[Ajie16]

好吧，谢谢

[Angelo-scut]

你好，请问是tensorflow的可以么？该怎么转？我在做seq2seq的LSTM模型部署，，pytorch转不过去，想请教一下