add point finetune: update config of lr_scheduler

Mark-ZhouWX · Mark-ZhouWX · commit 51c481e99da2 · 2024-01-15T11:44:59.000+08:00
diff --git a/official/cv/segment-anything/configs/coco_box_finetune.yaml b/official/cv/segment-anything/configs/coco_box_finetune.yaml
@@ -22,7 +22,7 @@ optimizer:
   group_param:
 
   lr_scheduler:
-    type: segment_anything.optim.scheduler.SAMDynamicDecayLR
+    type: segment_anything.optim.scheduler.sam_dynamic_decay_lr
     learning_rate: 8e-6
     warmup_steps: 250
     decay_steps: [ 60000, 86666 ]
diff --git a/official/cv/segment-anything/configs/flare_box_finetune.yaml b/official/cv/segment-anything/configs/flare_box_finetune.yaml
@@ -23,7 +23,7 @@ optimizer:
   group_param:
 
   lr_scheduler:
-    type: segment_anything.optim.scheduler.SAMDynamicDecayLR
+    type: segment_anything.optim.scheduler.sam_dynamic_decay_lr
     learning_rate: 8e-6
     warmup_steps: 250
     decay_steps: [ 60000, 86666 ]
diff --git a/official/cv/segment-anything/configs/profile_coco_box_finetune.yaml b/official/cv/segment-anything/configs/profile_coco_box_finetune.yaml
@@ -23,7 +23,7 @@ optimizer:
   group_param:
 
   lr_scheduler:
-    type: segment_anything.optim.scheduler.SAMDynamicDecayLR
+    type: segment_anything.optim.scheduler.sam_dynamic_decay_lr
     learning_rate: 8e-6
     warmup_steps: 250
     decay_steps: [ 60000, 86666 ]
diff --git a/official/cv/segment-anything/configs/profile_flare_box_finetune.yaml b/official/cv/segment-anything/configs/profile_flare_box_finetune.yaml
@@ -23,7 +23,7 @@ optimizer:
   group_param:
 
   lr_scheduler:
-    type: segment_anything.optim.scheduler.SAMDynamicDecayLR
+    type: segment_anything.optim.scheduler.sam_dynamic_decay_lr
     learning_rate: 8e-6
     warmup_steps: 250
     decay_steps: [ 60000, 86666 ]
diff --git a/official/cv/segment-anything/configs/sa1b_box_finetune.yaml b/official/cv/segment-anything/configs/sa1b_box_finetune.yaml
@@ -22,7 +22,7 @@ optimizer:
   group_param:
 
   lr_scheduler:
-    type: segment_anything.optim.scheduler.SAMDynamicDecayLR
+    type: segment_anything.optim.scheduler.sam_dynamic_decay_lr
     learning_rate: 8e-6
     warmup_steps: 250
     decay_steps: [ 60000, 86666 ]
diff --git a/official/cv/segment-anything/configs/sa1b_text_finetune_blip2.yaml b/official/cv/segment-anything/configs/sa1b_text_finetune_blip2.yaml
@@ -22,7 +22,7 @@ optimizer:
   group_param:
 
   lr_scheduler:
-    type: segment_anything.optim.scheduler.SAMDynamicDecayLR
+    type: segment_anything.optim.scheduler.sam_dynamic_decay_lr
     learning_rate: 8e-6
     warmup_steps: 250
     decay_steps: [ 60000, 86666 ]
diff --git a/official/cv/segment-anything/configs/sa1b_text_finetune_clip.yaml b/official/cv/segment-anything/configs/sa1b_text_finetune_clip.yaml
@@ -22,7 +22,7 @@ optimizer:
   group_param:
 
   lr_scheduler:
-    type: segment_anything.optim.scheduler.SAMDynamicDecayLR
+    type: segment_anything.optim.scheduler.sam_dynamic_decay_lr
     learning_rate: 8e-6
     warmup_steps: 250
     decay_steps: [ 60000, 86666 ]
