LSTM analysing onnx model error: TOOL ERROR: list index out of range

[Joker_han]

Hi,

I'm trying to test a model(Shallownet+LSTM) on the STM32F4, but the tool fails during analysis:

ST Edge AI Core v2.1.0-20194

TOOL ERROR: list index out of range

 

I ran the tool with debug enabled: export DEBUG=2

And I observed in the log:

Created input transpose layer: _lstm_LSTM_output_0_forward_in_transpose
Error in execution of pass type(OPT_TRANSPOSE_LSTM) id(80)

 

The model and code are included in the attachment.

How can I fix the issue ?

Thanks,

Joker_Han

 

[Julian E.]

Hello @Joker_han,

 

It is a bug on our side. Sorry for the inconvenience.

I opened an internal ticket to fix it.

 

As of today, LSTM and GRUs are badly supported. In the next update, work should have been included, but it was reported to fix more urgent issues.

I think we can expect improvements regarding these kinds of layers in the 2.3 (nothing for sure though)

 

I'll update you when I get any news.

Have a good day,

Julian