CUDA out of memory error yolov4 for coco 2017 dataset

AlexeyAB / Yolo_mark

GUI for marking bounded boxes of objects in images for training neural network Yolo v3 and v2

https://github.com/AlexeyAB/darknet

The Unlicense

1.8k stars 680 forks source link

CUDA out of memory error yolov4 for coco 2017 dataset #207

Open kumaramiit opened 3 years ago

kumaramiit commented 3 years ago

Hi, I am getting "cuda out of memory" error for yolov4 training with coco2017 datasets my cfg is: 1 [net] 2 batch=64 3 subdivisions=32 4 # Training 5 width=416 6 height=416 Gfx card details: gtx 5000 8 cores I started with subdivision 8, 16, 32 all giving same error Any help is appreciated

thegamercoder19 commented 3 years ago

Hello, this means your GPU doesn't have any memory for COCO with 2 images in subdivision. Use subdivision=64.