YOLOv4自定义模型训练:final.weights与last.weights哪个更优?
Darknet训练YOLOv4:last.weights与final.weights的区别及选择
两者核心区别
- last.weights:训练过程中每完成一次迭代就会更新的权重文件,记录的是训练停止时(无论正常结束还是中途中断)最后一轮的模型状态。如果训练中途意外中断,后续可以加载这个文件继续训练。
- final.weights:只有当训练跑完所有预设迭代次数后才会生成的文件——它本质是最后一轮迭代的
last.weights的复制版,参数完全一致。
该选哪一个?
- 如果训练正常完成所有迭代:两个文件的权重参数没有任何区别,任选其一即可。
- 如果训练中途中断:只有
last.weights是最新的模型状态,要么用它直接做推理,要么加载它继续完成剩余训练。
额外说明
你猜测final.weights是最优的best.weights,这是错误的。Darknet里要生成最优权重,需要在模型配置文件中设置save_best=1,程序会根据验证集性能指标(比如mAP)自动保存表现最好的模型,对应的文件才是best.weights。
内容的提问来源于stack exchange,提问作者Nathan
相关产品推荐
相关产品推荐

