blob: 92167152698c13e99f9296cb18165098d7196cff (
plain)
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}"
|
