_base_ = './deeplabv3_r101-d8_4xb2-40k_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.)