From 32da7b347c42190c81c13afc07803a2041c3302c Mon Sep 17 00:00:00 2001
From: shahules786 <shahules786@gmail.com>
Date: Sun, 16 Oct 2022 17:40:29 +0530
Subject: [PATCH] demucs + vctk 3 sec

---
 enhancer/cli/train.py                                  | 4 ++--
 enhancer/cli/train_config/config.yaml                  | 2 +-
 enhancer/cli/train_config/dataset/Vctk.yaml            | 6 +++---
 enhancer/cli/train_config/hyperparameters/default.yaml | 1 +
 enhancer/cli/train_config/trainer/default.yaml         | 2 +-
 5 files changed, 8 insertions(+), 7 deletions(-)

diff --git a/enhancer/cli/train.py b/enhancer/cli/train.py
index 398fa2b..abeccc8 100644
--- a/enhancer/cli/train.py
+++ b/enhancer/cli/train.py
@@ -47,9 +47,9 @@ def main(config: DictConfig):
     callbacks.append(checkpoint)
     if parameters.get("Early_stop", False):
         early_stopping = EarlyStopping(
-            monitor="val_loss",
+            monitor=f"valid_{parameters.get('EarlyStopping_metric','loss')}",
             mode=direction,
-            min_delta=0.0,
+            min_delta=parameters.get("EarlyStopping_delta", 0.00),
             patience=parameters.get("EarlyStopping_patience", 10),
             strict=True,
             verbose=False,
diff --git a/enhancer/cli/train_config/config.yaml b/enhancer/cli/train_config/config.yaml
index 9c464ac..8d0ab14 100644
--- a/enhancer/cli/train_config/config.yaml
+++ b/enhancer/cli/train_config/config.yaml
@@ -1,6 +1,6 @@
 defaults:
   - model : Demucs
-  - dataset : DNS-2020
+  - dataset : Vctk
   - optimizer : Adam
   - hyperparameters : default
   - trainer : default
diff --git a/enhancer/cli/train_config/dataset/Vctk.yaml b/enhancer/cli/train_config/dataset/Vctk.yaml
index a4707bf..a703c78 100644
--- a/enhancer/cli/train_config/dataset/Vctk.yaml
+++ b/enhancer/cli/train_config/dataset/Vctk.yaml
@@ -1,10 +1,10 @@
 _target_: enhancer.data.dataset.EnhancerDataset
 name : vctk
 root_dir : /scratch/c.sistc3/DS_10283_2791
-duration : 1.5
+duration : 3
 sampling_rate: 16000
-batch_size: 256
-valid_size : 0.05
+batch_size: 128
+valid_size : 0.025
 
 files:
   train_clean : clean_trainset_28spk_wav
diff --git a/enhancer/cli/train_config/hyperparameters/default.yaml b/enhancer/cli/train_config/hyperparameters/default.yaml
index 56f0713..fc99710 100644
--- a/enhancer/cli/train_config/hyperparameters/default.yaml
+++ b/enhancer/cli/train_config/hyperparameters/default.yaml
@@ -4,4 +4,5 @@ lr : 0.001
 ReduceLr_patience : 10
 ReduceLr_factor : 0.5
 min_lr : 0.00
+early_stop : True
 EarlyStopping_factor : 10
diff --git a/enhancer/cli/train_config/trainer/default.yaml b/enhancer/cli/train_config/trainer/default.yaml
index d4917e8..36a7912 100644
--- a/enhancer/cli/train_config/trainer/default.yaml
+++ b/enhancer/cli/train_config/trainer/default.yaml
@@ -23,7 +23,7 @@ limit_test_batches: 1.0
 limit_train_batches: 1.0
 limit_val_batches: 1.0
 log_every_n_steps: 100
-max_epochs: 250
+max_epochs: 300
 max_time: 00:47:00:00
 min_epochs: 1
 min_steps: null