clearml/examples/frameworks/jsonargparse/pytorch_lightning_cli_old.yml
2023-06-12 15:14:13 +03:00

14 lines
348 B
YAML

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: train_loss
mode: min
max_epochs: 3