mirrors/mmsegmentation
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmsegmentation.git synced 2025-06-03 22:03:48 +08:00
Code Issues Projects Releases Wiki Activity

Merge branch 'xiexinchen/fix_optim_config' into 'refactor_dev'

Browse Source 
[Fix] Fix mae/dpt optimizer config error

See merge request openmmlab-enterprise/openmmlab-ce/mmsegmentation!58
This commit is contained in:
zhengmiao 2022-07-07 11:04:10 +00:00
commit a63f77d249
2 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
configs/dpt/dpt_vit-b16_512x512_160k_ade20k.py
View File

@ -9,6 +9,7 @@ model = dict(data_preprocessor=data_preprocessor)
# in backbone
optim_wrapper = dict(
_delete_=True,
type='OptimWrapper',
optimizer=dict(
type='AdamW', lr=0.00006, betas=(0.9, 0.999), weight_decay=0.01),

1
configs/mae/upernet_mae-base_fp16_8x2_512x512_160k_ade20k.py
View File

@ -25,6 +25,7 @@ model = dict(
test_cfg=dict(mode='slide', crop_size=(512, 512), stride=(341, 341)))
optim_wrapper = dict(
_delete_=True,
type='OptimWrapper',
optimizer=dict(
type='AdamW', lr=1e-4, betas=(0.9, 0.999), weight_decay=0.05),