在mmseg框架下是默认用iters进行迭代训练的,但是可以自行设定使用epoch进行训练。
循环控制器指的是训练, 验证和测试时的执行流程, 在配置文件里面使用 train_cfg, val_cfg 和 test_cfg 来构建这些流程。
例如,使用基于迭代次数的训练循环 (IterBasedTrainLoop) 去训练 80,000 个迭代次数, 并且每 8,000 iteration 做一次验证, 可以如下设置:
train_cfg = dict(type='IterBasedTrainLoop', max_iters=80000, val_interval=8000)
在这里就可以调整,例如需要使用epoch进行训练,调整配置的类别信息。所有类别都在mmengine/runner/loops.py中。整体文件可以在官网查看:mmengine/mmengine/runner/loops.py at main · open-mmlab/mmengine · GitHub
mmengine/runner/loops.py中的EpochBased

最低0.47元/天 解锁文章
2046

被折叠的 条评论
为什么被折叠?



