From a3e488f101a2cd1f7b7f29aa807af2a253177cad Mon Sep 17 00:00:00 2001
From: shahules786 <shahules786@gmail.com>
Date: Wed, 2 Nov 2022 10:47:03 +0530
Subject: [PATCH] config

---
 enhancer/cli/train_config/config.yaml                  | 4 ++--
 enhancer/cli/train_config/dataset/DNS-2020.yaml        | 2 +-
 enhancer/cli/train_config/hyperparameters/default.yaml | 2 +-
 enhancer/cli/train_config/trainer/default.yaml         | 2 +-
 enhancer/data/dataset.py                               | 2 +-
 5 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/enhancer/cli/train_config/config.yaml b/enhancer/cli/train_config/config.yaml
index c0b2cf6..9c464ac 100644
--- a/enhancer/cli/train_config/config.yaml
+++ b/enhancer/cli/train_config/config.yaml
@@ -1,6 +1,6 @@
 defaults:
-  - model : WaveUnet
-  - dataset : Vctk
+  - model : Demucs
+  - dataset : DNS-2020
   - optimizer : Adam
   - hyperparameters : default
   - trainer : default
diff --git a/enhancer/cli/train_config/dataset/DNS-2020.yaml b/enhancer/cli/train_config/dataset/DNS-2020.yaml
index 3a1976b..159ecf7 100644
--- a/enhancer/cli/train_config/dataset/DNS-2020.yaml
+++ b/enhancer/cli/train_config/dataset/DNS-2020.yaml
@@ -1,5 +1,5 @@
 _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
diff --git a/enhancer/cli/train_config/hyperparameters/default.yaml b/enhancer/cli/train_config/hyperparameters/default.yaml
index 5f36268..c7a2672 100644
--- a/enhancer/cli/train_config/hyperparameters/default.yaml
+++ b/enhancer/cli/train_config/hyperparameters/default.yaml
@@ -3,6 +3,6 @@ metric : [stoi,pesq,si-sdr]
 lr : 0.003
 ReduceLr_patience : 10
 Early_stop : False
-ReduceLr_factor : 0.1
+ReduceLr_factor : 0.5
 min_lr : 0.0000001
 EarlyStopping_patience : 10
diff --git a/enhancer/cli/train_config/trainer/default.yaml b/enhancer/cli/train_config/trainer/default.yaml
index b6c4441..b5a474c 100644
--- a/enhancer/cli/train_config/trainer/default.yaml
+++ b/enhancer/cli/train_config/trainer/default.yaml
@@ -25,7 +25,7 @@ limit_val_batches: 1.0
 log_every_n_steps: 50
 max_epochs: 200
 max_steps: -1
-max_time: null
+max_time: 00:47:00:00
 min_epochs: 1
 min_steps: null
 move_metrics_to_cpu: False
diff --git a/enhancer/data/dataset.py b/enhancer/data/dataset.py
index d6ab415..2444c7f 100644
--- a/enhancer/data/dataset.py
+++ b/enhancer/data/dataset.py
@@ -133,7 +133,7 @@ class TaskDataset(pl.LightningDataModule):
             speaker_index = rng.choice(possible_indices)
             possible_indices.remove(speaker_index)
             speaker_name = all_speakers[speaker_index]
-            print(f"Selected f{speaker_name} for valid")
+            print(f"Selected {speaker_name} for valid")
             file_indices = [
                 i
                 for i, file in enumerate(data)