Relative Content

Tag Archive for pythonkerasmemorymemory-leaks

Keras memory leak when using custom DataGenerator

For the past 3 weeks I’ve been searching nonstop for a solution to this problem, when training a LSTM model with a custom DataGenerator, Keras ends up using all my RAM memory. The context of the project is to predict sleep stages, in this script, its expected to paralelyze 15 different participants with its 10 folds (10 train and 10 validation), and in a following phase test with its respective partition.
Having said that, this is the LSTM Network I’m currently using: