mmdetection修改epoch?或者训练完,怎么增加epoch继续训练?

问题:mmdetection想修改epoch;或者mmdetection训练完,但训练结果不够好,怎么增加epoch继续训练?

mmdetection版本  3.0.0
PyTorch  1.11.0
Python  3.8(ubuntu20.04)
Cuda  11.3

解决办法:

情况1:完全还没开始训练,想修改epoch。
  • 打开./configs/faster_rcnn/faster-rcnn_r50_fpn_1x_coco.py(这里根据个人需要自己选择打开文件哈),看到训练计划配置文件为./_base_/schedules/schedule_1x.py,记下这个文件路径

  • 打开刚才的训练计划配置文件./_base_/schedules/schedule_1x.py,标红框的几个地方需要修改

    ①max_epoches、end是最大的epoch数,改成自己想要的epoch数

    ②milestones是学习率下降的地方,[8,11]表示学习率在第8、11个epoch下降,随着总epoch数的改变,milestone也需要改变。学习了网上一些博主的做法,采用等比改变,也就是如果epoch总数为12

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值