search

sanghoon / pva-faster-rcnn

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

training pvanet in multiple gpu? #45

Open hengck23 opened 7 years ago

hengck23 commented 7 years ago

Is it possible to train pvanet in a single machine with multiple gpus?

According to "https://github.com/rbgirshick/py-faster-rcnn/issues/107", python layer makes it not possible to use multiple gpu.

However, this paper "Feature Pyramid Networks for Object Detection" (https://arxiv.org/abs/1612.03144) suggests that RPN framework can be trained using synchronized SGD training on 8 GPUs (using caffe2)

I am currently woking on COCO dataset for object detection with pvanet. Any hint to speed training using multiple GPU machine? Or will training on GPU cluster helps?

songjmcn commented 7 years ago

I think you will rewrite python layer using C++ and cuda.The python can not use multiple GPU.

hengck23 commented 7 years ago

mxnet faster rcnn (and also mxnet ssd, mxnet yolo) can support multi-gpu in single machine and multiple gpu machines. see: https://github.com/precedenceguo/mx-rcnn/issues/42

there is converter to convert mxnet model to caffe model and vice verus.