From f66a5236e196564ebcd61eb2944f1adf8b9bedbe Mon Sep 17 00:00:00 2001
From: shahules786 <shahules786@gmail.com>
Date: Tue, 11 Oct 2022 21:54:47 +0530
Subject: [PATCH] Revert "demucs"

This reverts commit d415bb0c59ab46e8a9fb141398ceb06f3805998a.
---
 enhancer/cli/train_config/config.yaml          | 2 +-
 enhancer/cli/train_config/dataset/Vctk.yaml    | 4 ++--
 enhancer/cli/train_config/trainer/default.yaml | 2 +-
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/enhancer/cli/train_config/config.yaml b/enhancer/cli/train_config/config.yaml
index 8d0ab14..c0b2cf6 100644
--- a/enhancer/cli/train_config/config.yaml
+++ b/enhancer/cli/train_config/config.yaml
@@ -1,5 +1,5 @@
 defaults:
-  - model : Demucs
+  - model : WaveUnet
   - dataset : Vctk
   - optimizer : Adam
   - hyperparameters : default
diff --git a/enhancer/cli/train_config/dataset/Vctk.yaml b/enhancer/cli/train_config/dataset/Vctk.yaml
index c38de61..5c19320 100644
--- a/enhancer/cli/train_config/dataset/Vctk.yaml
+++ b/enhancer/cli/train_config/dataset/Vctk.yaml
@@ -1,9 +1,9 @@
 _target_: enhancer.data.dataset.EnhancerDataset
 name : vctk
 root_dir : /scratch/c.sistc3/DS_10283_2791
-duration : 2.0
+duration : 1.0
 sampling_rate: 16000
-batch_size: 32
+batch_size: 64
 
 files:
   train_clean : clean_trainset_28spk_wav
diff --git a/enhancer/cli/train_config/trainer/default.yaml b/enhancer/cli/train_config/trainer/default.yaml
index b0ff02c..f50cae1 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: 100
+max_epochs: 500
 max_time: null
 min_epochs: 1
 min_steps: null