BobLiu20/YOLOv3_PyTorch

大佬您好,如何支持多卡训练?

Opened this issue · 3 comments

我原来一直在根据 https://github.com/eriklindernoren/PyTorch-YOLOv3 这个来做训练、预测,但是一直没有解决多卡训练的问题,我发现单纯的添加:
model = nn.DataParallel(model, device_ids=args.device)
model.cuda()
这两行,并不能为训练速度带来提升,我看您的代码有支持多卡训练,能帮忙解答下吗?感谢!

请问你最后解决了嘛,我也需要基于这个repo实现并行训练呢,求交流!!

现在一般没人用这个了,你可以用这个:https://github.com/ultralytics/yolov5

您好,请问您实现了在yolov3上用apex,求教呢