训练YOLOv5模型过程记录 1、运行train.py报错 pytorch gpu出错问题:RuntimeError: CUDA out of memory. 报错原因:GPU的显存不足,不能容纳1batch的数据量 解决方法:batch-size改小点 epochs改小点使训练快点结束 batch-size:多少数据打包成一个batch epochs:训练多少轮 (注:我的目的是让代码跑起来所以缩短训练时间) 训练成功的亚子