Object-Detection-Knowledge-Distillation icon indicating copy to clipboard operation
Object-Detection-Knowledge-Distillation copied to clipboard

Published 20 hours ago • verified publisher icon SsisyphusTao

RuntimeError: CUDA error: out of memory

Open LiYunJamesPhD opened this issue 4 years ago • 2 comments

Hi,

I always got the run time error after I ran python eval.py mbv2 --trained_model=checkpoints/student_mbv2_500_3934.pth. Does anyone know how to solve this error? Thanks.

LiYunJamesPhD avatar Jan 21 '21 04:01 LiYunJamesPhD

Did you try to reduce the batch size?

EdenBelouadah avatar Feb 02 '22 16:02 EdenBelouadah

@EdenBelouadah Thanks. No, I haven't. I will try it.

LiYunJamesPhD avatar Feb 02 '22 19:02 LiYunJamesPhD