I don’t hold any reference to outputs outside of the function.
What I would like is to use as much GPU memory as possible and not just only 10%. As I have access to a gtx 1080.
when my memory usage is around 90% my training goes quite fast, and when it drops it lslow down so much the process.
Thanks for the link I will take a look