File size: 251 Bytes
412c852 |
1 2 3 4 5 6 7 8 |
_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.)
|