ayush939 opened this issue 2 years ago · 0 comments
I am facing memory leak issue due to the gradient calculation of the noise_batch tensor. Is there any specific PyTorch version required to avoid this issue?