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.21k stars 260 forks source link

cuda out of memory #42

Open mouawiya opened 2 years ago

mouawiya commented 2 years ago

i've tried the code on a computer with graphic card GTX 1050 Ti and i still got the Cuda out of memory error, when i tried to lowed the batch size, it worked for a moment then stopped as it is showed in the picture below image

tanay-o commented 1 year ago

Hey! got the same error, try reducing the batch size when you train your dataset with the --batch_size parameter (default 4), that worked for me.