YOLOv5训练：epoch从0变1时，GPU显存消耗突然翻倍

[问题描述]训练YOLOv5模型，Epoch从0变1时，GPU显存消耗突然翻倍，例如：从3.54G 突然增长到8.08G

Epoch gpu_mem GIoU obj cls total targets img_size
0/299 3.54G 0.0874 0.0744 0.081 0.2428 10 640: 100%|
Epoch gpu_mem GIoU obj cls total targets img_size
1/299 8.08G 0.07261 0.0743 0.06651 0.2134 37 640
Epoch gpu_mem GIoU obj cls total targets img_size
2/299 8.08G 0.06912 0.06453 0.06019 0.1938 211 640

Epoch=0,gpu_mem=3.38G Epoch=1,gpu_mem=6.84G

[解决方案] 这是因为训练完毕后，执行Validation导致的显存翻倍。在训练时，加入参数 --noval即可。训练命令如下：

python train.py --img 640 --batch-size 32 --epochs 3 --data coco.yaml --weights yolov5s.pt --optimizer Adam --workers 6 --noval --nosav

GPU显存消耗不翻倍

参考链接：https://github.com/ultralytics/yolov5/issues/610