defaults:
  - model_checkpoint
  - model_summary
  - rich_progress_bar
  - _self_

model_checkpoint:
  dirpath: ${paths.output_dir}/checkpoints
  filename: "best-step-{step:06d}"
  monitor: "val/loss"
  mode: "min"
  save_last: True
  save_weights_only: False
  save_top_k: 0
  auto_insert_metric_name: False

safetensors:
  _target_: src.callbacks.safetensors_callback.SafetensorsCallback
  cleanup_orphan_safetensors: True

# early_stopping:
#   monitor: "val/loss"
#   patience: 100
#   mode: "min"

model_summary:
  max_depth: 1

device_stats:
  _target_: lightning.pytorch.callbacks.DeviceStatsMonitor

visualization:
  _target_: src.callbacks.visualization_callback.VisualizationCallback
  num_samples: 4