_base_ = './pspnet_r101-d8_512x1024_80k_cityscapes.py'
optim_wrapper = dict(
_delete_=True,
type='AmpOptimWrapper',
optimizer=dict(type='SGD', lr=0.01, momentum=0.9, weight_decay=0.0005),
loss_scale=512.)