summaryrefslogtreecommitdiff
path: root/training/conf/callbacks/early_stopping.yaml
blob: 4cd5aa12f2c493d5d4dc7ce4b7d6a13896ebbb0c (plain)
1
2
3
4
5
6
early_stopping:
  _target_: pytorch_lightning.callbacks.EarlyStopping
  monitor: "val/loss" # name of the logged metric which determines when model is improving
  patience: 16 # how many epochs of not improving until training stops
  mode: "min" # can be "max" or "min"
  min_delta: 0 # minimum change in the monitored metric needed to qualify as an improvement