DPACMAN / configs /callbacks /default.yaml
svincoff's picture
training works
29899b4
defaults:
- model_checkpoint
- early_stopping
- model_summary
- _self_
model_checkpoint:
dirpath: ${paths.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: 100
mode: "min"
model_summary:
max_depth: -1