search

ruotianluo / pytorch-faster-rcnn

pytorch1.0 updated. Support cpu test and demo. (Use detectron2, it's a masterpiece)
MIT License
1.82k stars 475 forks source link

Slower than tf-faster-rcnn #130

Open rnsandeep opened 5 years ago

rnsandeep commented 5 years ago

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.

karanchawla commented 5 years ago

Are you referring to training times or during prediction?

rnsandeep commented 5 years ago

Training time.