christmassa 2021-04-09 19:54 采纳率: 50%
浏览 160
已采纳

求助各位大佬!!! ylov5训练的多进程问题

看到网上的文章说,把多进程的文件放入

if __name__ == '__main__':的里面,但是默认就在这里面 还是会出现这个问题

 

  • 写回答

1条回答 默认 最新

  • 爱晚乏客游 2021-04-09 22:40
    关注

    yolov5自带的多进程是加载数据集用的,你想要加速得多显卡。

    parser.add_argument('--workers', type=int, default=6, help='maximum number of dataloader workers')
    #这里修改default 的数值就是多线程加载数据集的。但是要注意一定要小于cpu核心数,超过就报错。
    
     parser.add_argument('--device', default='', help='cuda device, i.e. 0 or 0,1,2,3 or cpu')
    #这句才是多显卡的,dafault="0,1,2,3"这么写就是四路显卡训练
    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论

报告相同问题?