Extremely slow when compute mAP

[zerteen]

Hi,

Your code is gorgeous that helps me a lot.
After I tried this code, I found it very slow when computing mAP, which is on CPU. I've tried to change it on GPU but got the out of memory error.
Do you have any suggests to accelerate this process?

Thanks.

[rayLemond]

try smaller BatchSize perhaps