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