diff --git a/projects/CO-DETR/configs/codino/co_dino_5scale_r50_lsj_8xb2_3x_coco.py b/projects/CO-DETR/configs/codino/co_dino_5scale_r50_lsj_8xb2_3x_coco.py index 9a9fc34f680..cec5411d61d 100644 --- a/projects/CO-DETR/configs/codino/co_dino_5scale_r50_lsj_8xb2_3x_coco.py +++ b/projects/CO-DETR/configs/codino/co_dino_5scale_r50_lsj_8xb2_3x_coco.py @@ -1,4 +1,4 @@ _base_ = ['co_dino_5scale_r50_lsj_8xb2_1x_coco.py'] -param_scheduler = [dict(milestones=[30])] +param_scheduler = [dict(type='MultiStepLR', milestones=[30])] train_cfg = dict(max_epochs=36) diff --git a/projects/CO-DETR/configs/codino/co_dino_5scale_swin_l_16xb1_3x_coco.py b/projects/CO-DETR/configs/codino/co_dino_5scale_swin_l_16xb1_3x_coco.py index c2fce29b98b..465b600decf 100644 --- a/projects/CO-DETR/configs/codino/co_dino_5scale_swin_l_16xb1_3x_coco.py +++ b/projects/CO-DETR/configs/codino/co_dino_5scale_swin_l_16xb1_3x_coco.py @@ -2,5 +2,5 @@ # model settings model = dict(backbone=dict(drop_path_rate=0.6)) -param_scheduler = [dict(milestones=[30])] +param_scheduler = [dict(type='MultiStepLR', milestones=[30])] train_cfg = dict(max_epochs=36)