update config

Author: Page0526

configs/callbacks/default.yaml

@@ -1,6 +1,6 @@
 defaults:
   - model_checkpoint
-  # - early_stopping
+  - early_stopping
   - model_summary
   - rich_progress_bar
   - _self_
@@ -13,10 +13,10 @@ model_checkpoint:
   save_last: True
   auto_insert_metric_name: False
 
-# early_stopping:
-#   monitor: "val/loss"
-#   patience: 100
-#   mode: "min"
+early_stopping:
+  monitor: "val/loss"
+  patience: 10
+  mode: "min"
 
 model_summary:
   max_depth: -1

configs/experiment/modl.yaml

@@ -19,7 +19,7 @@ seed: 12345
 
 trainer:
   min_epochs: 10
-  max_epochs: 50
+  max_epochs: 20
   gradient_clip_val: 0.5
   num_sanity_val_steps: 0
 
@@ -39,3 +39,9 @@ logger:
     name: "${now:%Y-%m-%d}_${now:%H-%M-%S}"
     # id: null # pass correct id to resume experiment! (if ckpt_path is not null)
 
+callbacks:
+  model_checkpoint:
+    monitor: "train/loss"
+
+  early_stopping:
+    monitor: "train/loss"