CUDA out of memory

[hellowordo]

Hi,Jia lian;
When I run train_net.py with the default settings and reached 22000 iterations, it reported an error of "cuda out of memory". May I ask what the reason is? (running on a single 48G GPU)

[kooWZ]

It's because CenterNet2 has a GPU memory leaking problem. You can follow xingyizhou/CenterNet2@a63c1d6 to fix it.