search

mitmul / chainer-faster-rcnn

Object Detection with Faster R-CNN in Chainer
MIT License
288 stars 87 forks source link

multi-gpu training? #17

Closed hengck23 closed 7 years ago

hengck23 commented 7 years ago

Chainer can support multiple gpus training (in a single machine). Does this version of faster-rcnn support multi-gpu?

mitmul commented 7 years ago

@hengck23 Sorry for late reply. Now it doesn't support multi-GPU training, but it's not so difficult to change the training code to support multi-GPU I think. Well, Hi, we've released the cleaned version of Faster R-CNN inference & training codes in ChainerCV: https://github.com/pfnet/chainercv . So, this repository has been actually deprecated. I'll consider to add multi-GPU training codes for Faster R-CNN there, so could you try the ChainerCV repo instead of this code? And please throw an issue when you face a problem :) Thank you!