Complex-YOLOv4-Pytorch icon indicating copy to clipboard operation
Complex-YOLOv4-Pytorch copied to clipboard

Published 20 hours ago • verified publisher icon maudzung

cuda out of memory

Open mouawiya opened this issue 3 years ago • 1 comments

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

mouawiya avatar May 18 '22 09:05 mouawiya

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.

tanay-o avatar Aug 04 '22 11:08 tanay-o