pytorch-faster-rcnn icon indicating copy to clipboard operation
pytorch-faster-rcnn copied to clipboard

Slower than tf-faster-rcnn

Open rnsandeep opened this issue 6 years ago • 2 comments

This code is taking more time to train than the tf-faster-rcnn. When I am running at 1800 scale this code is taking 2.9 sec /iter as compared to 1.7sec/iter in tf-faster-rcnn. Also going out of memory at 1800 input scale ( my gpu had 16gb of gpu memory) with this code, whereas tf-faster-rcnn isn't.

rnsandeep avatar Feb 06 '19 10:02 rnsandeep

Are you referring to training times or during prediction?

karanchawla avatar Feb 11 '19 22:02 karanchawla

Training time.

rnsandeep avatar Feb 12 '19 04:02 rnsandeep