Emptying cache for GPU memory leakage after every epoch#34

Open

luis-guerraf wants to merge 1 commit intosail-sg:mainfrom

luis-guerraf:main

Commits on May 4, 2022

I observed GPU memory leakage after every epoch, I couldn't find the root cause but emptying the cache fixed it.
luis-guerraf
committed