原本我以为在darnet中因为一些问题中断yolo模型的训练后,再利用之前中断的weights来重新训练是会在这个weights的轮次的基础上再加上cfg文件上的轮次,但今天我对一个已经完成cfg上轮次训练的a模型,再一次用一样的命令行命令进行训练,想要在这个cfg轮次的训练上再加一次cfg的轮次训练,不同的是weights改为已经训练好的a模型weights,但命令行运行后立刻说saving weights to … ,而且并没有报错。这貌似是weights中保存有上次完成的轮次训练信息,一旦和cfg中的轮次信息相同,则不再进行训练。
那么也就是说,在darknet中,无论如何中断训练,只要重新进行命令行命令,且weights改为上一次结束时的得到的weights,那么总的模型训练轮次是不会改变的。而像我这种想要对已经训练好的模型(轮次与cfg文件上的轮次一致),想要进一步添加训练轮次以继续训练,则必须修改cfg文件上的轮次信息。
yolo darknet模型训练中中断训练后再继续训练的问题
最新推荐文章于 2024-09-02 20:58:16 发布
博客讨论了在Darknet中如何使用预训练权重继续模型训练的问题。当试图以相同cfg轮次加载已完成训练的模型权重时,Darknet不会继续训练,而是直接保存权重,除非修改cfg文件的轮次信息。这意味着要增加训练轮次,必须更新cfg文件。
摘要由CSDN通过智能技术生成