From d99fd0eb61ff4d969f15ac4a05db0c209c209dc9 Mon Sep 17 00:00:00 2001
From: shahules786 <shahules786@gmail.com>
Date: Sun, 16 Oct 2022 11:13:44 +0530
Subject: [PATCH 1/2] fix duration estimation

---
 enhancer/models/model.py | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/enhancer/models/model.py b/enhancer/models/model.py
index aa0199b..dc5219d 100644
--- a/enhancer/models/model.py
+++ b/enhancer/models/model.py
@@ -115,17 +115,23 @@ class Model(pl.LightningModule):
             self.dataset.setup(stage)
             print(
                 "Total train duration",
-                self.dataset.train_dataloader().dataset.__len__() / 60,
+                self.dataset.train_dataloader().dataset.__len__()
+                * self.dataset.duration
+                / 60,
                 "minutes",
             )
             print(
                 "Total validation duration",
-                self.dataset.val_dataloader().dataset.__len__() / 60,
+                self.dataset.val_dataloader().dataset.__len__()
+                * self.dataset.duration
+                / 60,
                 "minutes",
             )
             print(
                 "Total test duration",
-                self.dataset.test_dataloader().dataset.__len__() / 60,
+                self.dataset.test_dataloader().dataset.__len__()
+                * self.dataset.duration
+                / 60,
                 "minutes",
             )
             self.dataset.model = self

From dab7e73d53a8357db3afe3b870a184d589aa6cc1 Mon Sep 17 00:00:00 2001
From: shahules786 <shahules786@gmail.com>
Date: Sun, 16 Oct 2022 11:14:13 +0530
Subject: [PATCH 2/2] DNS 2020

---
 enhancer/cli/train_config/dataset/DNS-2020.yaml | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/enhancer/cli/train_config/dataset/DNS-2020.yaml b/enhancer/cli/train_config/dataset/DNS-2020.yaml
index 3bd0e67..09a14fb 100644
--- a/enhancer/cli/train_config/dataset/DNS-2020.yaml
+++ b/enhancer/cli/train_config/dataset/DNS-2020.yaml
@@ -1,12 +1,12 @@
 _target_: enhancer.data.dataset.EnhancerDataset
-root_dir : /Users/shahules/Myprojects/enhancer/datasets/vctk_test
+root_dir : /Users/shahules/Myprojects/MS-SNSD
 name : dns-2020
-duration : 1.0
+duration : 2.0
 sampling_rate: 16000
 batch_size: 32
+valid_size: 0.05
 files:
-  root_dir : /Users/shahules/Myprojects/enhancer/datasets/vctk_test
-  train_clean : clean_test_wav
-  test_clean : clean_test_wav
-  train_noisy : clean_test_wav
-  test_noisy : clean_test_wav
+  train_clean : CleanSpeech_training
+  test_clean : CleanSpeech_training
+  train_noisy : NoisySpeech_training
+  test_noisy : NoisySpeech_training