YOLOv3_PyTorch icon indicating copy to clipboard operation
YOLOv3_PyTorch copied to clipboard

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

Open yangbisheng2009 opened this issue 5 years ago • 3 comments

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

yangbisheng2009 avatar Aug 21 '19 06:08 yangbisheng2009

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

genqiaolynn avatar Jan 12 '21 06:01 genqiaolynn

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

yangbisheng2009 avatar Jan 12 '21 06:01 yangbisheng2009

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

genqiaolynn avatar Jan 21 '21 14:01 genqiaolynn