early stopping enabled

2026-05-22 11:29:01 +02:00
parent 56cdf38d8c
commit a68e31facc
1 changed files with 3 additions and 3 deletions
@@ -26,7 +26,7 @@ mask:
 optimization:
  optimizer: adamw # 'adamw', 'sgd', or 'lars'
  freeze_weights: true # true for linear probing, false for full fine-tuning
-  epochs: 50
+  epochs: 300
  lr: 0.01
  weight_decay: 5.0e-4
  use_cosine_schedule: false
@@ -45,8 +45,8 @@ optimization:
  # - lr: 0.01 | 0.05 | 0.001
  # - weight_decay: 5.0e-4 | 0.0
  early_stopping:
-    enabled: false
-    patience: 10
+    enabled: true
+    patience: 15
    min_delta: 1.0e-4
    min_epochs: 50
    restore_best_weights: true