Hi!
I countered the same problem.And I can’t find a proper way to show my memory usage after every train epoch iteration.
Could u give me your code sample about how to print the mem usage after every iter?