Add learning rate monitor, save 10 best epoch models

PTG-Kitware · Nov 15, 2024 · 119689e · 119689e
1 parent ec3b095
commit 119689e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/configs/callbacks/default.yaml b/configs/callbacks/default.yaml
@@ -3,6 +3,7 @@ defaults:
   - early_stopping
   - model_summary
   - rich_progress_bar
+  - learning_rate_monitor
   - plot_metrics
   - _self_
 
@@ -11,11 +12,10 @@ defaults:
 model_checkpoint:
   dirpath: ${paths.output_dir}/checkpoints
   filename: "epoch_{epoch:03d}"
-  # monitor: "val/loss"
-  # mode: "min"
   monitor: "val/f1"
   mode: "max"
   save_last: True
+  save_top_k: 10 # save k best models (determined by above metric)
   auto_insert_metric_name: False
 
 early_stopping:

diff --git a/configs/callbacks/learning_rate_monitor.yaml b/configs/callbacks/learning_rate_monitor.yaml
@@ -0,0 +1,4 @@
+learning_rate_monitor:
+  _target_: pytorch_lightning.callbacks.LearningRateMonitor
+  logging_interval: epoch
+  log_momentum: false