YOLO训练中断解决方法、训练结束后增加/减少训练次数epoch方法

1. 训练中断,继续训练

训练因外界原因中断:

方法一:修改train.py文件中的resume参数,从中断前的最后一个epoch开始训练。

YOLO训练中断解决方法、训练结束后增加/减少训练次数epoch方法_第1张图片

 由于突然断电等原因,last.pt未保存或损坏时:

方法二:删除损坏的last.pt,直接将最后的一个best.pt或epcoch_xx.pt改为last.pt,继续训练。

YOLO训练中断解决方法、训练结束后增加/减少训练次数epoch方法_第2张图片

2.训练完成后,减少增加epoch次数

由于网络收敛或者未收敛,需提前结束训练或者继续训练时,可以修改epoch,然后resume修改为True继续训练。

修改训练过程产生的所有.yaml中的epochs参数

YOLO训练中断解决方法、训练结束后增加/减少训练次数epoch方法_第3张图片

 YOLO训练中断解决方法、训练结束后增加/减少训练次数epoch方法_第4张图片

 

 

你可能感兴趣的:(YOLO,深度学习)