training/conf/callbacks/checkpoint.yaml


1
2
3
4
5
6
7
8
9

model_checkpoint:
  _target_: pytorch_lightning.callbacks.ModelCheckpoint
  monitor: val/loss # name of the logged metric which determines when model is improving
  save_top_k: 1 # save k best models (determined by above metric)
  save_last: true # additionaly always save model from last epoch
  mode: min # can be "max" or "min"
  verbose: false
  dirpath: checkpoints/
  filename: {epoch:02d}