10 lines
347 B
Python
10 lines
347 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 = 160000
|
||
|
|
checkpoint_config = dict(by_epoch=False, interval=16000)
|
||
|
|
evaluation = dict(interval=16000, metric='mIoU')
|