Skip to content

default

defaults:
  - model_checkpoint
  - early_stopping
  - model_summary
  # - rich_progress_bar
  - _self_

model_checkpoint:
  dirpath: ${paths.time_output_dir}/checkpoints
  filename: "epoch_{epoch:03d}"
  monitor: "val/loss"
  mode: "min"
  save_last: True
  auto_insert_metric_name: False

early_stopping:
  monitor: "val/loss"
  patience: 3
  mode: "min"

model_summary:
  max_depth: -1