Skip to content
This repository has been archived by the owner on May 28, 2021. It is now read-only.

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

Open
yangbisheng2009 opened this issue Aug 21, 2019 · 3 comments
Open

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

yangbisheng2009 opened this issue Aug 21, 2019 · 3 comments

Comments

@yangbisheng2009
Copy link

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

@genqiaolynn
Copy link

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

@yangbisheng2009
Copy link
Author

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

@genqiaolynn
Copy link

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

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants