From 26a0b4814b8b1b3ca9836dd6412c765871e6b84d Mon Sep 17 00:00:00 2001 From: xiexinch Date: Wed, 6 Jul 2022 17:26:52 +0800 Subject: [PATCH] add _delete_=True to optim wrapper --- configs/dpt/dpt_vit-b16_512x512_160k_ade20k.py | 1 + configs/mae/upernet_mae-base_fp16_8x2_512x512_160k_ade20k.py | 1 + 2 files changed, 2 insertions(+) diff --git a/configs/dpt/dpt_vit-b16_512x512_160k_ade20k.py b/configs/dpt/dpt_vit-b16_512x512_160k_ade20k.py index 9f4618c3c..56b33d96b 100644 --- a/configs/dpt/dpt_vit-b16_512x512_160k_ade20k.py +++ b/configs/dpt/dpt_vit-b16_512x512_160k_ade20k.py @@ -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), diff --git a/configs/mae/upernet_mae-base_fp16_8x2_512x512_160k_ade20k.py b/configs/mae/upernet_mae-base_fp16_8x2_512x512_160k_ade20k.py index 6a410086b..b8eae174e 100644 --- a/configs/mae/upernet_mae-base_fp16_8x2_512x512_160k_ade20k.py +++ b/configs/mae/upernet_mae-base_fp16_8x2_512x512_160k_ade20k.py @@ -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),