diff --git a/enhancer/cli/train_config/dataset/Vctk.yaml b/enhancer/cli/train_config/dataset/Vctk.yaml
index ecfbb85..0c15c2a 100644
--- a/enhancer/cli/train_config/dataset/Vctk.yaml
+++ b/enhancer/cli/train_config/dataset/Vctk.yaml
@@ -4,7 +4,7 @@ root_dir : /scratch/c.sistc3/DS_10283_2791
 duration : 2
 stride : 1
 sampling_rate: 16000
-batch_size: 64
+batch_size: 32
 min_valid_minutes : 25
 files:
   train_clean : clean_trainset_28spk_wav
diff --git a/enhancer/cli/train_config/model/DCCRN.yaml b/enhancer/cli/train_config/model/DCCRN.yaml
index 3190391..ca98c35 100644
--- a/enhancer/cli/train_config/model/DCCRN.yaml
+++ b/enhancer/cli/train_config/model/DCCRN.yaml
@@ -4,7 +4,7 @@ sampling_rate : 16000
 complex_lstm : True
 complex_norm : True
 complex_relu : True
-masking_mode : True
+masking_mode : "E"
 
 encoder_decoder:
   initial_output_channels : 32