ubc-vision/COTR

Question

Closed this issue · 2 comments

Lucifer1002 commented

Hello, when running through the code with the pre-trained model, it appears that RuntimeError: CUDA out of memory. Tried to allocate 2.00 GiB (GPU 0; 7.79 GiB total capacity; 2.90 GiB already allocated; 1.83 GiB free; 4.80 GiB reserved in total by PyTorch).Is there any solution？For example, which parameters to adjust？

jiangwei221 commented

Have you tried to set the parameter to False?

COTR/COTR/inference/inference_helper.py

Line 19 in 682db5a

LARGE_GPU = True

Lucifer1002 commented

Thank you very much, your method is effective！