diff --git a/enhancer/cli/train_config/config.yaml b/enhancer/cli/train_config/config.yaml
index c0b2cf6..8d0ab14 100644
--- a/enhancer/cli/train_config/config.yaml
+++ b/enhancer/cli/train_config/config.yaml
@@ -1,5 +1,5 @@
 defaults:
-  - model : WaveUnet
+  - model : Demucs
   - dataset : Vctk
   - optimizer : Adam
   - hyperparameters : default
diff --git a/enhancer/cli/train_config/dataset/DNS-2020.yaml b/enhancer/cli/train_config/dataset/DNS-2020.yaml
index 3bd0e67..5949913 100644
--- a/enhancer/cli/train_config/dataset/DNS-2020.yaml
+++ b/enhancer/cli/train_config/dataset/DNS-2020.yaml
@@ -2,10 +2,9 @@ _target_: enhancer.data.dataset.EnhancerDataset
 root_dir : /Users/shahules/Myprojects/enhancer/datasets/vctk_test
 name : dns-2020
 duration : 1.0
-sampling_rate: 16000
+sampling_rate: 8000
 batch_size: 32
 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
diff --git a/enhancer/cli/train_config/dataset/Vctk.yaml b/enhancer/cli/train_config/dataset/Vctk.yaml
index 5c19320..f5fa316 100644
--- a/enhancer/cli/train_config/dataset/Vctk.yaml
+++ b/enhancer/cli/train_config/dataset/Vctk.yaml
@@ -2,8 +2,8 @@ _target_: enhancer.data.dataset.EnhancerDataset
 name : vctk
 root_dir : /scratch/c.sistc3/DS_10283_2791
 duration : 1.0
-sampling_rate: 16000
-batch_size: 64
+sampling_rate: 8000
+batch_size: 32
 
 files:
   train_clean : clean_trainset_28spk_wav
diff --git a/enhancer/cli/train_config/model/Demucs.yaml b/enhancer/cli/train_config/model/Demucs.yaml
index d91b5ff..40dba09 100644
--- a/enhancer/cli/train_config/model/Demucs.yaml
+++ b/enhancer/cli/train_config/model/Demucs.yaml
@@ -1,7 +1,7 @@
 _target_: enhancer.models.demucs.Demucs
 num_channels: 1
 resample: 2
-sampling_rate : 16000
+sampling_rate : 8000
 
 encoder_decoder:
   depth: 5
@@ -13,4 +13,4 @@ encoder_decoder:
 
 lstm:
   bidirectional: False
-  num_layers: 2
+  num_layers: 1