early stopping enabled
This commit is contained in:
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user