看到网上的文章说,把多进程的文件放入
if __name__ == '__main__':的里面,但是默认就在这里面 还是会出现这个问题
看到网上的文章说,把多进程的文件放入
if __name__ == '__main__':的里面,但是默认就在这里面 还是会出现这个问题
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"这么写就是四路显卡训练