Merge pull request #17 from ChEB-AI/feature/early_stopping

Add Early Stopping feature

Author: sfluegel05

.gitignore

@@ -160,4 +160,4 @@ cython_debug/
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
 #.idea/
 
-configs/
+# configs/ # commented as new configs can be added as a part of a feature

configs/training/default_trainer.yml

@@ -1,5 +1,5 @@
-min_epochs: 100
+# min_epochs: 100
 max_epochs: 100 
 default_root_dir: &default_root_dir logs
-logger: wandb_logger.yml
+logger: csv_logger.yml
 callbacks: default_callbacks.yml

configs/training/early_stop_callbacks.yml

@@ -0,0 +1,19 @@
+- class_path: chebai.callbacks.model_checkpoint.CustomModelCheckpoint
+  init_args:
+    monitor: val_micro-f1
+    mode: 'max'
+    filename: 'best_{epoch:02d}_{val_loss:.4f}_{val_macro-f1:.4f}_{val_micro-f1:.4f}'
+    every_n_epochs: 1
+    save_top_k: 5
+- class_path: chebai.callbacks.model_checkpoint.CustomModelCheckpoint
+  init_args:
+    filename: 'per_{epoch:02d}_{val_loss:.4f}_{val_macro-f1:.4f}_{val_micro-f1:.4f}'
+    every_n_epochs: 5
+    save_top_k: -1
+- class_path: lightning.pytorch.callbacks.early_stopping.EarlyStopping
+  init_args:
+    monitor: "val_loss_epoch"
+    min_delta: 0.0
+    patience: 3
+    verbose: False
+    mode: "min"