From 75aa54b9d7f0dc6428f92dc9a44bd8a14c1019a9 Mon Sep 17 00:00:00 2001
From: shahules786 <shahules786@gmail.com>
Date: Tue, 15 Nov 2022 14:55:24 +0530
Subject: [PATCH] config

---
 enhancer/cli/train_config/config.yaml                  | 2 +-
 enhancer/cli/train_config/dataset/DNS-2020.yaml        | 5 ++---
 enhancer/cli/train_config/hyperparameters/default.yaml | 4 ++--
 enhancer/cli/train_config/mlflow/experiment.yaml       | 2 +-
 enhancer/cli/train_config/trainer/default.yaml         | 2 +-
 5 files changed, 7 insertions(+), 8 deletions(-)

diff --git a/enhancer/cli/train_config/config.yaml b/enhancer/cli/train_config/config.yaml
index 9c464ac..aea9269 100644
--- a/enhancer/cli/train_config/config.yaml
+++ b/enhancer/cli/train_config/config.yaml
@@ -1,5 +1,5 @@
 defaults:
-  - model : Demucs
+  - model : DCCRN
   - dataset : DNS-2020
   - optimizer : Adam
   - hyperparameters : default
diff --git a/enhancer/cli/train_config/dataset/DNS-2020.yaml b/enhancer/cli/train_config/dataset/DNS-2020.yaml
index 823ad38..09e56a3 100644
--- a/enhancer/cli/train_config/dataset/DNS-2020.yaml
+++ b/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
diff --git a/enhancer/cli/train_config/hyperparameters/default.yaml b/enhancer/cli/train_config/hyperparameters/default.yaml
index 9b9bd7b..93db6aa 100644
--- a/enhancer/cli/train_config/hyperparameters/default.yaml
+++ b/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
diff --git a/enhancer/cli/train_config/mlflow/experiment.yaml b/enhancer/cli/train_config/mlflow/experiment.yaml
index b907f62..36af7ed 100644
--- a/enhancer/cli/train_config/mlflow/experiment.yaml
+++ b/enhancer/cli/train_config/mlflow/experiment.yaml
@@ -1,2 +1,2 @@
 experiment_name : shahules/enhancer
-run_name : Demucs-DNS
+run_name : DCCRN-DNS-15
diff --git a/enhancer/cli/train_config/trainer/default.yaml b/enhancer/cli/train_config/trainer/default.yaml
index 797d259..b23133f 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: 50
-max_epochs: 150
+max_epochs: 100
 max_steps: -1
 max_time: null
 min_epochs: 1