config

enhancer/cli/train_config/config.yaml

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

enhancer/cli/train_config/dataset/DNS-2020.yaml

@@ -1,8 +1,7 @@
 _target_: enhancer.data.dataset.EnhancerDataset
-root_dir : /scratch/c.sistc3/MS-SNSD/DNS30
+root_dir : /scratch/c.sistc3/MS-SNSD/DNS15
 name : dns-2020
-duration : 5.0
-stride : 1
+duration : 2
 sampling_rate: 16000
 batch_size: 32
 min_valid_minutes: 25.0

enhancer/cli/train_config/hyperparameters/default.yaml

@@ -1,6 +1,6 @@
-loss : mae
+loss : si-snr
 metric : [stoi,pesq]
-lr : 0.0003
+lr : 0.001
 ReduceLr_patience : 10
 Early_stop : False
 ReduceLr_factor : 0.5

enhancer/cli/train_config/mlflow/experiment.yaml

@@ -1,2 +1,2 @@
 experiment_name : shahules/enhancer
-run_name : Demucs-DNS
+run_name : DCCRN-DNS-15

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