大佬您好,如何支持多卡训练?
Opened this issue · 3 comments
yangbisheng2009 commented
我原来一直在根据 https://github.com/eriklindernoren/PyTorch-YOLOv3 这个来做训练、预测,但是一直没有解决多卡训练的问题,我发现单纯的添加:
model = nn.DataParallel(model, device_ids=args.device)
model.cuda()
这两行,并不能为训练速度带来提升,我看您的代码有支持多卡训练,能帮忙解答下吗?感谢!
genqiaolynn commented
请问你最后解决了嘛,我也需要基于这个repo实现并行训练呢,求交流!!
yangbisheng2009 commented
现在一般没人用这个了,你可以用这个:https://github.com/ultralytics/yolov5
genqiaolynn commented
您好,请问您实现了在yolov3上用apex,求教呢