多GPU训练Yolov3

[tencent]

多GPU训练Yolov3

[mileistone]

We don't support multi gpus now. It's easy, you can implement it in 10 minutes.

[tonysy]

Could you share how to revise the code to achieve multi-gpu train? I found it difficult to implement.

[yutaizhou]

Can you share a guide on multi gpu implementation? Or give some good resources?

[gongh4]

First, update the cfg set gpus:"0,1"
Then, in _voc_train.py, insert the code after
if self.cuda:
net.cuda()
as
net.net = torch.nn.DataParallel(net.net)

Then, you can train using multi-gpu