Source: src/olm/train/callbacks/early_stopping_cb.py:1
Early stopping callback to prevent overfitting.
Classes
EarlyStoppingCallback(patience: int = 5, min_delta: float = 0.0)
Bases: olm.train.trainer.trainer.TrainerCallback
Source: src/olm/train/callbacks/early_stopping_cb.py:8
Callback to stop training early if validation loss doesn't improve.
Parameters
patience: Number of validation checks to wait for improvement.min_delta: Minimum change in validation loss to qualify as improvement.
Methods
on_step_end(self, trainer, step: int, loss: float) -> None
Source: src/olm/train/callbacks/early_stopping_cb.py:24
Check for early stopping after each step.