cancel
Showing results for 
Search instead for 
Did you mean: 

How do I run an RNN series model file?

IAN_K
Associate

Hello

I'm considering running an LSTM(RNN-based model) model file, on the STN32N657 NPU.

However, I understand that the NPU is CNN-based and therefore cannot run on it.

Is there a way to run an LSTM model file on the NPU?

I understand that theoretically, it's possible to convert an LSTM to a 1D CNN or TCN.

Has anyone converted an LSTM model to a 1D CNN or TCN and tested it on the NPU?

Is there an example file available?

Thank you.

 

1 REPLY 1
Julian E.
ST Employee

Hi @IAN_K,

 

To run a LSTM on NPU you need to decompose it (unroll) and run a timestep per inference.

 

You cannot convert a STM into 1D / TCN. You can use CNN 1D or TCN to work with timeseries as you would with a LSTM, but you cannot convert it.

 

Have a good day,

Julian


In order to give better visibility on the answered topics, please click on 'Accept as Solution' on the reply which solved your issue or answered your question.