search

BobLiu20 / YOLOv3_PyTorch

Full implementation of YOLOv3 in PyTorch
624 stars 191 forks source link

大佬您好，如何支持多卡训练？ #47

Open yangbisheng2009 opened 5 years ago

yangbisheng2009 commented 5 years ago

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

genqiaolynn commented 3 years ago

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

yangbisheng2009 commented 3 years ago

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

genqiaolynn commented 3 years ago

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