diff --git a/configs/fpn_neck.py b/configs/fpn_neck.py index 49a1b4b..64d3011 100644 --- 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 index 44de580..b576787 100644 --- 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 index 10cfbaf..0251211 100644 --- 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 index 9abc5fb..63599c8 100644 --- 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 index 9b30f5f..545fdb2 100644 --- 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 index 81b33b1..8c89ef1 100644 --- 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 +