yolo darknet模型训练中中断训练后再继续训练的问题

博客讨论了在Darknet中如何使用预训练权重继续模型训练的问题。当试图以相同cfg轮次加载已完成训练的模型权重时,Darknet不会继续训练,而是直接保存权重,除非修改cfg文件的轮次信息。这意味着要增加训练轮次,必须更新cfg文件。
摘要由CSDN通过智能技术生成

原本我以为在darnet中因为一些问题中断yolo模型的训练后,再利用之前中断的weights来重新训练是会在这个weights的轮次的基础上再加上cfg文件上的轮次,但今天我对一个已经完成cfg上轮次训练的a模型,再一次用一样的命令行命令进行训练,想要在这个cfg轮次的训练上再加一次cfg的轮次训练,不同的是weights改为已经训练好的a模型weights,但命令行运行后立刻说saving weights to … ,而且并没有报错。这貌似是weights中保存有上次完成的轮次训练信息,一旦和cfg中的轮次信息相同,则不再进行训练。
那么也就是说,在darknet中,无论如何中断训练,只要重新进行命令行命令,且weights改为上一次结束时的得到的weights,那么总的模型训练轮次是不会改变的。而像我这种想要对已经训练好的模型(轮次与cfg文件上的轮次一致),想要进一步添加训练轮次以继续训练,则必须修改cfg文件上的轮次信息。

评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值