mirror of
https://github.com/open-mmlab/mmsegmentation.git
synced 2025-06-03 22:03:48 +08:00
8 lines
251 B
Python
8 lines
251 B
Python
_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.)
|