From ca6797c3f17ff546588fd8ca2992b988d2d63e55 Mon Sep 17 00:00:00 2001
From: shahules786 <shahules786@gmail.com>
Date: Thu, 10 Nov 2022 10:41:07 +0530
Subject: [PATCH] config

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

diff --git a/enhancer/cli/train_config/config.yaml b/enhancer/cli/train_config/config.yaml
index 9c464ac..77877c6 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
+  - model : DCCRN
+  - 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 208098a..ecfbb85 100644
--- a/enhancer/cli/train_config/dataset/Vctk.yaml
+++ b/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
diff --git a/enhancer/cli/train_config/hyperparameters/default.yaml b/enhancer/cli/train_config/hyperparameters/default.yaml
index 0392f55..0bcb9e0 100644
--- a/enhancer/cli/train_config/hyperparameters/default.yaml
+++ b/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
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