You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
{{ message }}
This repository has been archived by the owner on May 28, 2021. It is now read-only.
我原来一直在根据 https://github.com/eriklindernoren/PyTorch-YOLOv3 这个来做训练、预测,但是一直没有解决多卡训练的问题,我发现单纯的添加:
model = nn.DataParallel(model, device_ids=args.device)
model.cuda()
这两行,并不能为训练速度带来提升,我看您的代码有支持多卡训练,能帮忙解答下吗?感谢!
The text was updated successfully, but these errors were encountered: