【yolov5 v6.0】中断以后重新训练,增加epochs,冻结层

文章讨论了如何在训练中断后使用resume参数继续训练,注意不要手动复制权重文件以免路径混乱。增加epochs以追求模型收敛,但需留意重新训练可能导致loss突然增加。使用tensorboard监控训练过程,并提到通过冻结模型部分层来进行微调的策略。

中断以后重新训练

有个resume的参数,将default从False改成True,然后他就会自己去找最新的权重继续训练了。
然后有个需要注意的点就是,不要为了想要备份最新的权重,然后把它复制一份出来,这样子,好像会造成路径的错乱,然后报错了很久哈哈哈(最后把复制出来的weights的文件夹删掉才好的)

增加训练的epochs

就是训练完了以后发现还没有收敛,想要继续增加epochs。我做的改动

  1. 修改resume参数
  2. 把epochs的参数修改成你需要的次数
  3. 在train文件夹里面加入语句,ckpt[‘epoch’] = 上一次训练的最后一个的次数
    在这里插入图片描述
    如果没有改的话可能会报错,大概的意思是说已经训练完了,不需要再继续训练了。
  4. 我还改了/runs/exp3中的opt.yaml文件中的epochs的个数,修改成了300,但是不知道这个有没有用处

训练的结果

tensorboard的使用

  1. 先在对应的环境下安装这个库
  2. 然后输入tensorboard --logdir=【event文件所在的文件夹路径】
  3. 然后在对应的端口打开
    在这里插入图片描述
  4. 如果打开网页以后没有响应,或者找不到网页的话,那么就需要在xshell中把对应的端口弄到本地服务器上面。
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值