mmocr/configs/_base_/schedules/schedule_sgd_600e.py
Xinyu Wang 28c9e460d5
[Enhancement] Add SynthText dataset config (#983)
* add synthtext config; add db pretrain config; move checkpoint config to schedule

* add db_r50 synthtext config

* fix r50 eval iter

* fix lint

* fix
2022-05-05 12:57:07 +08:00

9 lines
304 B
Python

# optimizer
optimizer = dict(type='SGD', lr=1e-3, momentum=0.99, weight_decay=5e-4)
optimizer_config = dict(grad_clip=None)
# learning policy
lr_config = dict(policy='step', step=[200, 400])
# running settings
runner = dict(type='EpochBasedRunner', max_epochs=600)
checkpoint_config = dict(interval=100)