search

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

💎1MB lightweight face detection model (1MB轻量级人脸检测模型)
MIT License
7.17k stars 1.54k forks source link

CUDA out of memory #241

Open KimMinGyu-ay opened 3 years ago

KimMinGyu-ay commented 3 years ago

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 3 years ago

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