mmdet里workers_per_gpu和sampers_per_gpu的作用

背景：

最近用mmdet的时候发现一个问题，在pipeline里进行一些随机操作（例如随机裁剪）的时候，设定一个随机种子random seed，理论上每次随机操作的结果都不同，但是实际上发现会有2张图的操作结果是一样的。

本来以为是batch_size的问题，就去修改了sampers_per_gpu 。结果发现实际上是workers_per_gpu的问题。因此就来好好研究下这俩个参数的作用和意义。

实际上科班的应该对进程比较熟悉，但是也考虑到有很多像我一样非科班的小白，可能对进程workers不是很理解，故此记录下，也欢迎大佬交流指正

这俩个参数具体出现在configs文件里

sampers_per_gpu

这个参数会传入在mmdet/datasets/builder.py里

if dist:# When model is :obj:`DistributedDataParallel`,# `batch_size` of :obj:`dataloader` is the# number of training samples on each GPU.batch_size = samples_per_gpunum_workers = workers_per_gpuelse:# When model is obj:`DataParallel`# the batch size is samples on all the GPUSbatch_size = num_gpus * samples_per_gpunum_workers = num_gpus * workers_per_gpu

可以看到sampers_per_gpu就是控制batch_size的，决定了dataloader一次从dataset里取几个样本（几张图）。

注意这里单卡和多卡不一样，简单说sampers_per_gpu是决定每张卡取多少sampers，总的batch_size要乘以你用的gpu显卡数目

workers_per_gpu

这个参数决定读取数据时每个gpu分配的线程数 。

数据加载时每个 GPU 使用的子进程（subprocess）数目。

因此如果workers=2，那就会把一个相同的随机种子seed同时作用于两个samper，因此就会导致出现两个分支（进程），就会产生不同组的图做的随机操作完全一样的情况。