can't run because of low Vram

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.24k stars 256 forks source link

can't run because of low Vram #22

Open Jsijon opened 3 years ago

Jsijon commented 3 years ago

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?

wangx1996 commented 3 years ago

@Jsijon hello，did you solve the problem?

mouawiya commented 3 years ago

@Jsijon hello，did you solve the problem?

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