ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-06-26 18:16:07 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Support pytorch-lightning CLI >=2.0

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

5
examples/frameworks/jsonargparse/pytorch_lightning_cli.yml
View File

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