目录
一、情况:
情况:
- 1.训练过程中中断了,继续训练
- 2.训练完了,但是未收敛,在这个基础上,还想用这个权重、学习率等参数继续训练多一些批次
1. 训练过程中中断了,继续训练
断电、或者什么原因中断了,比如设定epoch为200,但是在90这里中断了,想从断点这里继续 训练到 200个epoch
直接在trainds.py设置参数–resume的defualt=True即可(在代码 517行附近)
parser.add_argument('--resume', nargs='?', const=True, default=True, help='resume most recent training')
2.训练完原有epoch,发现没有收敛,想继续训练
训练完原有epoch后,但还继续训练,比如设置epoch为300,已经训练完了,但是没有完全收敛,想使用训练了300 epoch的权重继续训练200个epoch, 总共就是500个epoch
注意:不是直接在train.py设置weight参数为训练好的权重(last.pt),然后设置epoch为500,这

本文档介绍了在YoloV5/V7模型训练过程中如何进行断点续训和在已有epoch基础上继续训练。重点讲述了在train.py和trainds.py中设置参数,包括修改`--resume`、`--epochs`,以及调整学习率等关键步骤,以确保训练能够从已保存的权重继续,并达到预期的训练周期。
最低0.47元/天 解锁文章
1287

被折叠的 条评论
为什么被折叠?



