DAFNe

DAFNe copied to clipboard

RuntimeError: CUDA out of memory

11

While training the model on dota1.0, gpu memory keeps on increasing after every few iterations until it runs out of memory. I have tried both docker and without docker training....

keshavoct98

File missing

1

file "dafne.modeling.nms.nms.py" line 6 ->from poly_nms import poly_gpu_nms where can I find this poly_nms module? look forward to your answer

yangkk2019

Environment

8

Whats the python version you are using?

aikeet2927

188.0 58.0 200.0 67.0 159.0 117.0 147.0 108.0 harbor 0 323.0 159.0 331.0 170.0 293.0 203.0 286.0 194.0 harbor 0

sahilutekar