mirror of
https://github.com/clearml/clearml
synced 2025-01-31 09:07:00 +00:00
14 lines
343 B
YAML
14 lines
343 B
YAML
trainer:
|
|
callbacks:
|
|
- class_path: lightning.pytorch.callbacks.LearningRateMonitor
|
|
init_args:
|
|
logging_interval: epoch
|
|
- class_path: lightning.pytorch.callbacks.ModelCheckpoint
|
|
init_args:
|
|
filename: best
|
|
save_last: False
|
|
save_top_k: 1
|
|
monitor: loss
|
|
mode: min
|
|
max_epochs: 10
|