lyuwenyu / RT-DETR

[CVPR 2024] Official RT-DETR (RTDETR paddle pytorch), Real-Time DEtection TRansformer, DETRs Beat YOLOs on Real-time Object Detection. 🔥 🔥 🔥
Apache License 2.0
2.61k stars 303 forks source link

Memory leaks after evaluation #93

Open patriciamdr opened 1 year ago

patriciamdr commented 1 year ago

Hi!

I am observing CPU RAM memory leaks when training RT-DETR with PyTorch on multiple GPUs after the first evaluation. After that, the RAM usage increased from ~30GB to ~100GB, and continously increases further (and slows down training due to using swapfile). Do you have any ideas what could be the reason for this?

Thanks!

lyuwenyu commented 1 year ago

I rerun rtdetr-r50 on coco dataset, there is a similar phenomenon to what you said, but my machine has 600G ram. I don't know what problem caused it yet.

epoch RAM(G)
01     66
15     114
36     125
42     119