Hi, I tried to reproduce the training, but there is a memory leak. Can you provide a fixed version of the repo? Thanks Luca