search

YuwenXiong / py-R-FCN

R-FCN with joint training and python support
MIT License
1.05k stars 472 forks source link

Does this project support trained by multiple GPUs? #5

Open chungjin opened 8 years ago

chungjin commented 8 years ago

I wander whether your project can be trained by multiple GPUs? I have tried to use the command like --gpu 0,1,2 but it seems it does not support this way. But in Dai's paper, they use 8 GPUs to train the model.

zhangjiangqige commented 7 years ago

I don't think this is possible, since it used python layer, and caffe does not support multi-gpu if python layer is used

chungjin commented 7 years ago

I think so, so they must have used the C++ and Cuda to rewrite the python layer

bharatsingh430 commented 7 years ago

https://github.com/bharatsingh430/py-R-FCN-multiGPU

You can use this branch