early stopping enabled

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