Linzaer/Ultra-Light-Fast-Generic-Face-Detector-1MB

CUDA out of memory

KimMinGyu-ay opened this issue · 1 comments

KimMinGyu-ay commented

I encounter "CUDA out of memory. Tried to allocate 44.00 MiB (GPU 0; 4.00 GiB total capacity; 1.05 GiB already allocated; 102.40 KiB free; 1.07 GiB reserved in total by PyTorch)"

could you help me to solve id ?

i don't know how to solve this Erorr.....

kimtaehyeong commented

For learning, would you like to reduce the batch size?