mirror of
https://github.com/clearml/clearml
synced 2025-01-31 00:56:57 +00:00
14 lines
348 B
YAML
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
|