config

enhancer/cli/train_config/config.yaml

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

enhancer/cli/train_config/dataset/Vctk.yaml

@@ -4,7 +4,7 @@ root_dir : /scratch/c.sistc3/DS_10283_2791
 duration : 2
 stride : 1
 sampling_rate: 16000
-batch_size: 128
+batch_size: 64
 min_valid_minutes : 25
 files:
   train_clean : clean_trainset_28spk_wav

enhancer/cli/train_config/hyperparameters/default.yaml

@@ -1,7 +1,7 @@
-loss : mae
-metric : [si-sdr,stoi,pesq]
-lr : 0.0003
-ReduceLr_patience : 100
+loss : si-snr
+metric : [stoi,pesq]
+lr : 0.001
+ReduceLr_patience : 10
 Early_stop : False
 ReduceLr_factor : 0.1
 min_lr : 0.0000001

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: 50
-max_epochs: 150
+max_epochs: 100
 max_steps: -1
 max_time: null
 min_epochs: 1