问题: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