Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

training pvanet in multiple gpu? #45

Open
hengck23 opened this issue Dec 29, 2016 · 2 comments
Open

training pvanet in multiple gpu? #45

hengck23 opened this issue Dec 29, 2016 · 2 comments

Comments

@hengck23
Copy link

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

According to "rbgirshick#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
Copy link

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

@hengck23
Copy link
Author

mxnet faster rcnn (and also mxnet ssd, mxnet yolo) can support multi-gpu in single machine and multiple gpu machines. see: ijkguo/mx-rcnn#42

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants