libtorch1.2 adam cuda memory leaks, but sgd no leaks
can any one help me?
also occurs in RMSProp
Do you have a reproducible code snippet so that we could have a look?
please refer to :
there is no leaks in CPU.
and in GPU, 2M leaks every time .
please also refter to :
Thanks for the code and for opening an issue.
For the same of completeness: the issue is tracked here.