From 4e58df5e374e305f5715bfffc41f9d1cc6df5f14 Mon Sep 17 00:00:00 2001
From: shahules786 <shahules786@gmail.com>
Date: Thu, 10 Nov 2022 12:07:02 +0530
Subject: [PATCH] reduce bs

---
 enhancer/cli/train_config/dataset/Vctk.yaml | 2 +-
 enhancer/cli/train_config/model/DCCRN.yaml  | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

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