search

sanghoon / pva-faster-rcnn

Demo code for PVANet
https://arxiv.org/abs/1611.08588
Other
650 stars 241 forks source link

How to speed up training process? end2end? #65

Open user1103 opened 7 years ago

user1103 commented 7 years ago

Can we use pvanet with end2end (from faster-rcnn)? If so, how? What is the command like? I notice that it takes pretty long to train my model.

If its not possible, what can I do to speed up the training process?

I want to get good mAP, so I'd have to use the full model, not compressed.

This is my overall loss after 100k iterations: loss

Amocy-Wang commented 7 years ago
  1. There are lots of deep learning frameworks accessed by researchers and developers. Each framework has its own advantages, one of these features may be speed for some specific models. You can see some papers for more information about different frameworks.
  2. Use better GPUs or FPGAs. Most of frameworks support for distribution computing.