yolov5为什么默认要训练300次这么多

Question

mmdetection平台下的检测器12次就能训练完了,但是yolov5要训练300次这么多,我的gpu显存24g,yolov5能达到64个batchsize,但是在mmdetection下面16个batchsize就爆了,而且yolov5的每个epoch训练起来都好快,只要几分钟

Answer 1

mmdetection没试过，但是yolov5默认300多次是因为人家是在coco数据集上面训练的，而你看yolov5的主页，里面的各个模型的对比可以看出来，基本上都是300epoch的时候达到较好的效果，之后收敛速度变慢，效果不大。如果你训练自己的数据集，你看看loss,P,R和mAP，这些数据达到稳定之后基本上就可以停止了，并不需要300epoch。16个epoch就爆，具体没用过，我猜测估计应该是用空间换时间吧，或者哪里优化没弄好。