将works设置为0即可
将上面标记的改成0。再次运行。
出现RuntimeError: CUDA out of memory. Tried to allocate 26.00 MiB (GPU 0; 4.00 GiB total capacity; 2.03 GiB already allocated; 1.91 MiB free; 2.05 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF这个错误。
再将batch-size改成3
将上面的改成3即可,再次运行
还是出现以下情况的,将循环次数改成10即可
改成10,再次运行。
运行成功
另外,我建议大家使用yolov5.4版本的