Emptying cache for GPU memory leakage after every epoch

sail-sg / poolformer

PoolFormer: MetaFormer Is Actually What You Need for Vision (CVPR 2022 Oral)

https://arxiv.org/abs/2111.11418

Apache License 2.0

1.29k stars 117 forks source link

Open luis-guerraf opened 2 years ago

luis-guerraf commented 2 years ago

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.