From 1831dc201385a47d286acc804e9c172ed3514570 Mon Sep 17 00:00:00 2001
From: shahules786 <shahules786@gmail.com>
Date: Wed, 12 Oct 2022 17:28:04 +0530
Subject: [PATCH] config

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

diff --git a/enhancer/cli/train_config/dataset/Vctk.yaml b/enhancer/cli/train_config/dataset/Vctk.yaml
index 5c19320..df50da2 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: 64
+batch_size: 128
 
 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 55101de..92ae56a 100644
--- a/enhancer/cli/train_config/trainer/default.yaml
+++ b/enhancer/cli/train_config/trainer/default.yaml
@@ -22,8 +22,8 @@ limit_predict_batches: 1.0
 limit_test_batches: 1.0
 limit_train_batches: 1.0
 limit_val_batches: 1.0
-log_every_n_steps: 1
-max_epochs: 10
+log_every_n_steps: 50
+max_epochs: 3
 max_steps: null
 max_time: null
 min_epochs: 1