Fix one of the CO-DETR config files (#11325)

open-mmlab · Jan 3, 2024 · aeb4647 · aeb4647
1 parent 46b10b1
commit aeb4647
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/projects/CO-DETR/configs/codino/co_dino_5scale_swin_l_lsj_16xb1_3x_coco.py b/projects/CO-DETR/configs/codino/co_dino_5scale_swin_l_lsj_16xb1_3x_coco.py
@@ -2,5 +2,6 @@
 
 model = dict(backbone=dict(drop_path_rate=0.5))
 
-param_scheduler = [dict(milestones=[30])]
+param_scheduler = [dict(type='MultiStepLR', milestones=[30])]
+
 train_cfg = dict(max_epochs=36)