From 4cfe7a2463a2839e60acf49ee1ee3404304579c3 Mon Sep 17 00:00:00 2001
From: shahules786 <shahules786@gmail.com>
Date: Sat, 22 Oct 2022 12:07:45 +0530
Subject: [PATCH] Demus + DNS

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

diff --git a/enhancer/cli/train_config/config.yaml b/enhancer/cli/train_config/config.yaml
index 8d0ab14..9c464ac 100644
--- a/enhancer/cli/train_config/config.yaml
+++ b/enhancer/cli/train_config/config.yaml
@@ -1,6 +1,6 @@
 defaults:
   - model : Demucs
-  - dataset : Vctk
+  - 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 3d9177d..c42a52c 100644
--- a/enhancer/cli/train_config/dataset/DNS-2020.yaml
+++ b/enhancer/cli/train_config/dataset/DNS-2020.yaml
@@ -3,9 +3,9 @@ root_dir : /scratch/c.sistc3/MS-SNSD/DNS10
 name : dns-2020
 duration : 5.0
 sampling_rate: 16000
-batch_size: 64
-stride: 2
-valid_minutes: 30.0
+batch_size: 128
+stride: 1
+valid_minutes: 15.0
 files:
   train_clean : CleanSpeech_training
   test_clean : CleanSpeech_testing
diff --git a/enhancer/cli/train_config/trainer/default.yaml b/enhancer/cli/train_config/trainer/default.yaml
index 4fa4438..2d6422c 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: 100
-max_epochs: 400
+max_epochs: 200
 max_time: 00:47:00:00
 min_epochs: 1
 min_steps: null