mmocr/configs/_base_/schedules/schedule_adam_1e.py

7 lines
169 B
Python

# optimizer
optimizer = dict(type='Adam', lr=1e-3)
optimizer_config = dict(grad_clip=None)
# learning policy
lr_config = dict(policy='poly', power=0.9)
total_epochs = 1