10 lines
344 B
Python
10 lines
344 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, min_lr=1e-4, by_epoch=False)
|
||
|
|
# runtime settings
|
||
|
|
total_iters = 20000
|
||
|
|
checkpoint_config = dict(by_epoch=False, interval=2000)
|
||
|
|
evaluation = dict(interval=2000, metric='mIoU')
|