demucs + vctk 3 sec

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,

enhancer/cli/train_config/config.yaml

@@ -1,6 +1,6 @@
 defaults:
   - model : Demucs
-  - dataset : DNS-2020
+  - dataset : Vctk
   - optimizer : Adam
   - hyperparameters : default
   - trainer : default

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
+batch_size: 128
-valid_size : 0.05
+valid_size : 0.025
 
 files:
   train_clean : clean_trainset_28spk_wav

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

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