From 8a55a7764056acdd5d173dbfc9151929716f997c Mon Sep 17 00:00:00 2001 From: shahules786 Date: Thu, 13 Oct 2022 10:52:22 +0530 Subject: [PATCH] run 100 epochs --- enhancer/cli/train_config/dataset/Vctk.yaml | 2 +- enhancer/cli/train_config/hyperparameters/default.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 5905bc6..6bdc8c7 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 : 1.0 sampling_rate: 16000 batch_size: 128 -valid_size : 0.1 +valid_size : 0.05 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 e4f1429..b822ccc 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,stoi,pesq] +metric : [stoi,pesq,si-sdr] lr : 0.001 ReduceLr_patience : 5 ReduceLr_factor : 0.1