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

Question

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

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

 

Answer 1

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"这么写就是四路显卡训练