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

Published 20 hours ago • verified publisher icon maudzung

can't run because of low Vram

Open Jsijon opened this issue 5 years ago • 2 comments

RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 3.79 GiB total capacity; 2.48 GiB already allocated; 25.50 MiB free; 2.52 GiB reserved in total by PyTorch)

I tried to lower the batch_size to 1 in train_config but i still have not enough memory. is there any parameter to change so it becomes less memory hungry? what do you suggest?

Jsijon avatar Oct 16 '20 17:10 Jsijon

@Jsijon hello，did you solve the problem?

wangx1996 avatar Jan 04 '21 10:01 wangx1996

@Jsijon hello，did you solve the problem?

i have the same problem now, can you please help me if you have a solution

mouawiya avatar Sep 17 '21 14:09 mouawiya