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