search

maudzung / Complex-YOLOv4-Pytorch

The PyTorch Implementation based on YOLOv4 of the paper: "Complex-YOLO: Real-time 3D Object Detection on Point Clouds"
https://arxiv.org/pdf/1803.06199.pdf
GNU General Public License v3.0
1.24k stars 256 forks source link

Cuda out of memory, #32

Open mouawiya opened 3 years ago

mouawiya commented 3 years ago

hello when i tried to run the evaluate.py file , like it is mentioned in the Readme file, i always have this error, and i don't know how to fix it, mind you that i have 2Gb nvidia graphic card GTX 950m, when i tried to lower the batch size it worked, but i lowered it till it was equal 1 and the results were wrong, can you please help me image

YanyanDai commented 2 years ago

Change batch_size=2, num_work=2. It worked for me.