poolformer icon indicating copy to clipboard operation
poolformer copied to clipboard

Emptying cache for GPU memory leakage after every epoch

Open luis-guerraf opened this issue 2 years ago • 0 comments

Hi, I observed GPU memory leakage after every epoch, I couldn't find the root cause but emptying the cache fixed it, might be helpful.

luis-guerraf avatar May 04 '22 05:05 luis-guerraf