update

config/model_config/yolo_free_v2_config.py

@@ -65,7 +65,7 @@ yolo_free_v2_cfg = {
         'ema_decay': 0.9998,       # SGD: 0.9999;   AdamW: 0.9998
         'ema_tau': 2000,
         ## LR schedule
-        'scheduler': 'cos_linear',
+        'scheduler': 'linear',
         'lr0': 0.001,              # SGD: 0.01;     AdamW: 0.001
         'lrf': 0.01,               # SGD: 0.01;     AdamW: 0.01
         'warmup_momentum': 0.8,
@@ -135,7 +135,7 @@ yolo_free_v2_cfg = {
         'ema_decay': 0.9998,       # SGD: 0.9999;   AdamW: 0.9998
         'ema_tau': 2000,
         ## LR schedule
-        'scheduler': 'cos_linear',
+        'scheduler': 'linear',
         'lr0': 0.001,              # SGD: 0.01;     AdamW: 0.001
         'lrf': 0.01,               # SGD: 0.01;     AdamW: 0.01
         'warmup_momentum': 0.8,

train.py

@@ -157,7 +157,7 @@ def train():
     if args.eval_first and distributed_utils.is_main_process():
         # to check whether the evaluator can work
         model_eval = model_without_ddp
-        trainer.eval_one_epoch(model_eval)
+        trainer.eval(model_eval)
 
     ## Satrt Training
     trainer.train(model)