Add args.validation_interval to config. Use args.validation_interval to skip validation and stats logging at every epoch.

Author: Abdul-Mukit

rfdetr/config.py

@@ -130,6 +130,7 @@ class TrainConfig(BaseModel):
     ema_tau: int = 100
     lr_drop: int = 100
     checkpoint_interval: int = 10
+    validation_interval: int = 1
     warmup_epochs: float = 0.0
     lr_vit_layer_decay: float = 0.8
     lr_component_decay: float = 0.7

rfdetr/main.py

@@ -368,6 +368,11 @@ def lr_lambda(current_step: int):
 
                         utils.save_on_master(weights, checkpoint_path)
 
+            # Run validation and log stats every validation_interval
+            if (epoch + 1) % args.validation_interval != 0:
+                print(f"Skipping epoch {epoch} for evaluation. validation_interval = {args.validation_interval}")
+                continue
+
             with torch.inference_mode():
                 test_stats, coco_evaluator = evaluate(
                     model, criterion, postprocess, data_loader_val, base_ds, device, args=args
@@ -946,6 +951,7 @@ def populate_args(
     output_dir='output',
     dont_save_weights=False,
     checkpoint_interval=10,
+    validation_interval=1,
     seed=42,
     resume='',
     start_epoch=0,
@@ -1053,6 +1059,7 @@ def populate_args(
         output_dir=output_dir,
         dont_save_weights=dont_save_weights,
         checkpoint_interval=checkpoint_interval,
+        validation_interval=validation_interval,
         seed=seed,
         resume=resume,
         start_epoch=start_epoch,