Possible Memory Leak on CPU

[ptrblck]

Thanks for the code!
I could reproduce this behavior and the memory usage is growing ~1GB each epoch on my machine.
I’ll take a closer look at it.