Tencent / FaceDetection-DSFD

腾讯优图高精度双分支人脸检测器
Other
2.91k stars 728 forks source link

GPU out of memory #88

Open LiuZhenshun opened 2 years ago

LiuZhenshun commented 2 years ago

Hi, I tried to use 8G 2080-ti to evaluate your 152-resnet model, but the memory is not enough. Could you tell the minimum Memory to run your test code?

gkrisp98 commented 1 year ago

Hi, I am also facing the same problem. Also, I would like to ask you how did you manage to run the demo.py file without the error about the autograd function? I have rewrite some lines of code and I was wondering if the error for cuda appears due to the changes I made