Lighthead-RCNN-in-Pytorch0.4.1 icon indicating copy to clipboard operation
Lighthead-RCNN-in-Pytorch0.4.1 copied to clipboard

Published 20 hours ago • verified publisher icon TreB1eN

Training is Too Slow

Open mmaaz60 opened this issue 5 years ago • 1 comments

Hi @TreB1eN,

While training on COCO with my RTX 2080 Super GPU, it is taking around 20 hours per epoch which is to slow. I mean to run 30 epochs, I need to wait for around 25 days?

How can the training be accelerated? Thanks

mmaaz60 avatar May 12 '20 09:05 mmaaz60

it runs RPN in numpy so speed is slow, you can try using pytorch code to accelerate

lucasjinreal avatar May 12 '20 13:05 lucasjinreal