CUDA out of memory error yolov4 for coco 2017 dataset
kumaramiit opened this issue · comments
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
Hello, this means your GPU doesn't have any memory for COCO with 2 images in subdivision. Use subdivision=64.