15 lines
367 B
Python
15 lines
367 B
Python
# optimizer
|
|
optimizer = dict(type='SGD', lr=0.01, momentum=0.9, weight_decay=0.0005)
|
|
optimizer_config = dict()
|
|
# learning policy
|
|
lr_config = dict(
|
|
policy='poly',
|
|
power=0.9,
|
|
by_epoch=False,
|
|
)
|
|
# runtime settings
|
|
total_iters = 60000
|
|
checkpoint_config = dict(by_epoch=False, interval=4000)
|
|
evaluation = dict(interval=4000, metric='mIoU')
|
|
runner_type = 'iter'
|