RuntimeError: CUDA out of memory. Tried to allocate
python train.py --img 640 --batch 16 --epochs 5 --data coco128.yaml --weights yolov5s.pt把batchsize改小 python train.py --im
python train.py --img 640 --batch 16 --epochs 5 --data coco128.yaml --weights yolov5s.pt把batchsize改小 python train.py --im
问题如下: 1.问题如下: RuntimeError: CUDA error: device-side assert triggered CUDA kernel errors might be asynchron