解决方法一：

将batch_size改为1：

打开train.py文件将442行的 batch_size改为1，将459行num_work改为0；

parser.add_argument('--batch-size', type=int, default=4, help='total batch size for all GPUs, -1 for autobatch')
... 
parser.add_argument('--workers', type=int, default=0, help='max dataloader workers (per RANK in DDP mode)')

解决方法二：

采用CPU训练：

打开train.py文件将442行的  default='  '，改为 default='cpu'

parser.add_argument('--device', default='cpu', help='cuda device, i.e. 0 or 0,1,2,3 or cpu')