MTCNN-Tensorflow icon indicating copy to clipboard operation
MTCNN-Tensorflow copied to clipboard

Published 20 hours ago • verified publisher icon AITTSMD

在运行gen_hard_example.py的时候显存溢出

Open Mi-zo-re opened this issue 5 years ago • 1 comments

在运行到4900张图片的时候爆出了CUDA_ERROR_OUT_OF_MEMORY的错误，请问该怎么解决这个问题啊

Mi-zo-re avatar Apr 29 '19 06:04 Mi-zo-re

that sounds like you GPU cannot support the current mini-batch size. Go ahead and reduce the mini-batch size

RoboEvangelist avatar Apr 30 '19 15:04 RoboEvangelist