解决yolov5-5.0 RuntimeError: CUDA out of memory. Tried to allocate 56.00 MiB (GPU 0； 7.79 GiB total c

模型训练的时候，明明batch-size设置恰当，img-size设置恰当，但还是报了如下错误

运行train.py训练的时候报如上Error。

解决办法：如果batch-size设置较小值，img-size值调小也还是不行的话。将

parser.add_argument('--workers', type=int, default=8, help='maximum number of dataloader workers')

中的参数default=8改为default=0即可解决。