Support pytorch-lightning CLI >=2.0

This commit is contained in:
allegroai
2023-05-11 15:59:11 +03:00
parent 43f7c549fb
commit a0bc87ab5c
7 changed files with 170 additions and 124 deletions

View File

@@ -0,0 +1,12 @@
trainer:
callbacks:
- class_path: pytorch_lightning.callbacks.LearningRateMonitor
init_args:
logging_interval: epoch
- class_path: pytorch_lightning.callbacks.ModelCheckpoint
init_args:
filename: best
save_last: False
save_top_k: 1
monitor: loss
mode: min