finetune/neck experiments

Brilliant-B · Aug 9, 2023 · 705ae1c · 705ae1c
1 parent 8da2d25
commit 705ae1c
Show file tree

Hide file tree

Showing 6 changed files with 17 additions and 15 deletions.
diff --git a/configs/fpn_neck.py b/configs/fpn_neck.py
@@ -27,12 +27,12 @@
 lr_config = dict(
     policy='poly',
     warmup='linear',
-    warmup_iters=1500,
-    warmup_ratio=1e-6,
+    warmup_iters=80000, # 1500,
+    warmup_ratio=1e-11, # 1e-6,
     power=1.0,
     min_lr=0.0, 
     by_epoch=False,
 )
 
-checkpoint_config = dict(interval=2000,)
+checkpoint_config = dict(interval=5000,)
 evaluation = dict(interval=10000,)
diff --git a/configs/linear_neck.py b/configs/linear_neck.py
@@ -23,12 +23,13 @@
 lr_config = dict(
     policy='poly',
     warmup='linear',
-    warmup_iters=1500,
-    warmup_ratio=1e-6,
+    warmup_iters=40000,# 1500,
+    warmup_ratio=1e-11,# 1e-6,
     power=1.0, 
     min_lr=0.0, 
     by_epoch=False,
 )
 
-checkpoint_config = dict(interval=2000,)
+runner = dict(type='IterBasedRunner', max_iters=40000)
+checkpoint_config = dict(interval=100,)
 evaluation = dict(interval=10000,)
diff --git a/configs/sfp_neck.py b/configs/sfp_neck.py
@@ -28,12 +28,12 @@
 lr_config = dict(
     policy='poly',
     warmup='linear',
-    warmup_iters=1500,
-    warmup_ratio=1e-6,
+    warmup_iters=80000, # 1500,
+    warmup_ratio=1e-11, # 1e-6,
     power=1.0, 
     min_lr=0.0, 
     by_epoch=False,
 )
 
-checkpoint_config = dict(interval=2000,)
+checkpoint_config = dict(interval=5000,)
 evaluation = dict(interval=10000,)
diff --git a/test.sh b/test.sh
@@ -3,7 +3,7 @@
 GPU=0
 NAME="eva2_segmenter"
 CONFIGS="configs/linear_neck.py"
-CKPT="/hy-tmp/workbench/eva2_segmenter/train/neck_linear_finetune_0/iter_22000.pth"
+CKPT="/hy-tmp/workbench/eva2_segmenter/train/neck_linear_finetune_0/100.pth"
 WORK_DIR="/hy-tmp/workbench/${NAME}/eval/"
 SHOW_DIR="/hy-tmp/workbench/${NAME}/eval/inference/"
 

diff --git a/train.py b/train.py
@@ -283,7 +283,7 @@ def main(args, info, verbose=False):
 
 if __name__ == '__main__':
     args = parse_args()
-    neck_choices = ["fpn", "sfp", "linear"]
+    neck_choices = ["fpn", "sfp", "linear"][2:]
     # finetune_code = {0: no-freeze, 1: freeze EVA}
     for n in neck_choices:
         hyper_info = {

diff --git a/train.sh b/train.sh
@@ -4,10 +4,11 @@ GPU=0
 NAME="eva2_segmenter"
 WORK_DIR="/hy-tmp/workbench/${NAME}/train/"
 RESUME=""
-LOAD=""
+LOAD="/hy-tmp/workbench/eva2_segmenter/train/neck_linear_finetune_0/100.pth"
 
 python train.py --seed 0 --deterministic --gpu-ids ${GPU} \
  --work-dir ${WORK_DIR} \
-#  --resume-from ${RESUME} \
-#  --no-validate
-#  --load-from ${LOAD}
+# --resume-from ${RESUME} \
+# --load-from ${LOAD}
+# --no-validate
+