search

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.3k stars 117 forks source link

Emptying cache for GPU memory leakage after every epoch #34

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.