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 a1e1394..5c19320 100644
--- a/enhancer/cli/train_config/dataset/Vctk.yaml
+++ b/enhancer/cli/train_config/dataset/Vctk.yaml
@@ -3,7 +3,7 @@ name : vctk
 root_dir : /scratch/c.sistc3/DS_10283_2791
 duration : 1.0
 sampling_rate: 16000
-batch_size: 16
+batch_size: 64
 
 files:
   train_clean : clean_trainset_28spk_wav
diff --git a/enhancer/cli/train_config/hyperparameters/default.yaml b/enhancer/cli/train_config/hyperparameters/default.yaml
index 2966c20..c2da54e 100644
--- a/enhancer/cli/train_config/hyperparameters/default.yaml
+++ b/enhancer/cli/train_config/hyperparameters/default.yaml
@@ -1,5 +1,5 @@
 loss : mse
-metric : mse
+metric : [mse,pesq,stoi]
 lr : 0.001
 ReduceLr_patience : 10
 ReduceLr_factor : 0.1