From 8906496366332fb7a297c69d47f84c6d1f152374 Mon Sep 17 00:00:00 2001 From: shahules786 Date: Wed, 12 Oct 2022 10:49:00 +0530 Subject: [PATCH] waveunet 500 epochs --- enhancer/cli/train.py | 4 ++-- enhancer/cli/train_config/dataset/Vctk.yaml | 3 ++- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/enhancer/cli/train.py b/enhancer/cli/train.py index 38300fd..5e12142 100644 --- a/enhancer/cli/train.py +++ b/enhancer/cli/train.py @@ -48,10 +48,10 @@ def main(config: DictConfig): early_stopping = EarlyStopping( monitor="val_loss", mode=direction, - min_delta=0.0, + min_delta=1e-7, patience=parameters.get("EarlyStopping_patience", 10), strict=True, - verbose=False, + verbose=True, ) callbacks.append(early_stopping) diff --git a/enhancer/cli/train_config/dataset/Vctk.yaml b/enhancer/cli/train_config/dataset/Vctk.yaml index 5c19320..5905bc6 100644 --- a/enhancer/cli/train_config/dataset/Vctk.yaml +++ b/enhancer/cli/train_config/dataset/Vctk.yaml @@ -3,7 +3,8 @@ name : vctk root_dir : /scratch/c.sistc3/DS_10283_2791 duration : 1.0 sampling_rate: 16000 -batch_size: 64 +batch_size: 128 +valid_size : 0.1 files: train_clean : clean_trainset_28spk_wav