Arduino_TensorFlowLite_ESP32 icon indicating copy to clipboard operation
Arduino_TensorFlowLite_ESP32 copied to clipboard

Support for LSTM and Autoencoder

Open expeon07 opened this issue 3 years ago • 6 comments

Hi, I'm trying to implement a multivariate time series prediction model using LSTM, and an anomaly detection model using autoencoders. I have successfully converted them to a TFLite version and tried to upload them in my ESP32.

However I get the following error: Only 1 subgraph is currently supported.

Are LSTMs and anomaly detection supported by this library? If so, are there special configurations that I need to do?

Thanks!

expeon07 avatar Mar 23 '22 14:03 expeon07

I'm sorry. Since it is an old version, some functions cannot be used. It is recommended to use the latest version with ES-IDF.

https://www.tensorflow.org/lite/microcontrollers

I'm planning to upgrade, but I don't have time.

tanakamasayuki avatar Mar 24 '22 09:03 tanakamasayuki

@expeon07 could you please share with me the code for converting the LSTM model to TFLite? I have not succeeded in the conversion. Thank you

koulnodji6056 avatar Dec 02 '22 07:12 koulnodji6056

@expeon07 could you please share with me the code for converting the LSTM model to TFLite? I have not succeeded in the conversion. Thank you

Hi, I unrolled the LSTM layer by turning unroll=True (default to false)

expeon07 avatar Dec 07 '22 22:12 expeon07

@expeon07 please I am not getting it clear. Is that turning of unroll=True (default to false) was done during model training of during model conversion (TFliteConverter)? kindly, I request more explanation or a close orientation. NB: I don't have much experience in coding

koulnodji6056 avatar Dec 10 '22 04:12 koulnodji6056

@expeon07 please I am not getting it clear. Is that turning of unroll=True (default to false) was done during model training of during model conversion (TFliteConverter)? kindly, I request more explanation or a close orientation. NB: I don't have much experience in coding

Hi, I think it's during the model training. I cannot remember as I only tried it months ago. Have you tried doing either/both yourself? You can ask the developers in the TFlite repo as well

expeon07 avatar Dec 10 '22 10:12 expeon07

@expeon07 Noted with thanks

koulnodji6056 avatar Dec 10 '22 11:12 koulnodji6056