YOLOP icon indicating copy to clipboard operation
YOLOP copied to clipboard

change batch size

Open danielzhangau opened this issue 3 years ago • 1 comments

After reviewing source code and modified segmentation parts, training raised the following error: RuntimeError: CUDA out of memory. Tried to allocate 24.00 MiB (GPU 0; 10.89 GiB total capacity; 9.69 GiB already allocated; 25.56 MiB free; 9.73 GiB reserved in total by PyTorch) I changed batch size from 24 to 8, solved this issue. But what will be the effect of doing this?

danielzhangau avatar Nov 18 '21 06:11 danielzhangau

modify in ./lib/config/default.py at _C.TEST.BATCH_SIZE_PER_GPU

JamesLeeeeeee avatar Jul 15 '22 02:07 JamesLeeeeeee