AlphaBetaGamma96:

When you say “machine details” do you mean like hardware? Or software?


I think the used CPU and OS would be a good starter.



 AlphaBetaGamma96:

However, I’ve noticed that if I have N=8 (for the network input) the memory leak seems to go away and the memory usage just fluctuates around 0.5 Gb but if I have say N=12 the memory leak is present and increases by around 0.1GB per epoch.


Interesting findings, which might help debug it further.
Please let us know once you can share some code.