clearml/examples/frameworks/jsonargparse/pytorch_lightning_cli.yml

14 lines
349 B
YAML
Raw Permalink Normal View History

2022-03-29 08:00:33 +00:00
trainer:
callbacks:
2023-05-11 12:59:11 +00:00
- class_path: lightning.pytorch.callbacks.LearningRateMonitor
2022-03-29 08:00:33 +00:00
init_args:
logging_interval: epoch
2023-05-11 12:59:11 +00:00
- class_path: lightning.pytorch.callbacks.ModelCheckpoint
2022-03-29 08:00:33 +00:00
init_args:
filename: best
save_last: False
save_top_k: 1
2023-06-12 12:14:13 +00:00
monitor: train_loss
2022-03-29 08:00:33 +00:00
mode: min
2023-05-11 12:59:11 +00:00
max_epochs: 10