8 lines
255 B
Python
8 lines
255 B
Python
|
_base_ = './deeplabv3plus_r101-d8_4xb2-80k_cityscapes-512x1024.py'
|
||
|
optimizer = dict(type='SGD', lr=0.01, momentum=0.9, weight_decay=0.0005)
|
||
|
|
optim_wrapper = dict(
|
||
|
|
_delete_=True,
|
||
|
|
type='AmpOptimWrapper',
|
||
|
|
optimizer=optimizer,
|
||
|
|
loss_scale=512.)
|