princeton-vl/CornerNet-Lite

but i get another problem,even i set batch_size=1,it also raise RuntimeError: CUDA out of memory. Tried to allocate 6.75 MiB (GPU 0; 3.94 GiB total capacity; 1.50 GiB already allocated; 23.00 MiB free; 49.10 MiB cached),it is wired??

cqray1990 opened this issue · 1 comments

but i get another problem,even i set batch_size=1,it also raise RuntimeError: CUDA out of memory. Tried to allocate 6.75 MiB (GPU 0; 3.94 GiB total capacity; 1.50 GiB already allocated; 23.00 MiB free; 49.10 MiB cached),it is wired??

Maybe these actions can help you?
princeton-vl/CornerNet#34