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

add _delete_=True to optim wrapper

Browse Source
This commit is contained in:
xiexinch 2022-07-06 17:26:52 +08:00
parent d54f80c649
commit 26a0b4814b
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),