RuntimeError: CUDA out of memory
amitmane1 opened this issue · comments
amitmane1 commented
I'm getting RuntimeError for GPU even after I have set batch size to 1 instead of mentioned in repo 16.
Could you please guide me into this for any possible correction?
GPU is having capacity of 8GB
Thank you!
YOLO commented
if you use windows, set --n_cpu to 0