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
