From a417e226f3ad7c5f10dde428adbc16588b25ccc5 Mon Sep 17 00:00:00 2001 From: shahules786 Date: Mon, 10 Oct 2022 12:49:41 +0530 Subject: [PATCH] testrun for metrics --- enhancer/cli/train_config/config.yaml | 2 +- enhancer/cli/train_config/dataset/Vctk.yaml | 2 +- enhancer/cli/train_config/hyperparameters/default.yaml | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/enhancer/cli/train_config/config.yaml b/enhancer/cli/train_config/config.yaml index 8d0ab14..c0b2cf6 100644 --- a/enhancer/cli/train_config/config.yaml +++ b/enhancer/cli/train_config/config.yaml @@ -1,5 +1,5 @@ defaults: - - model : Demucs + - model : WaveUnet - dataset : Vctk - optimizer : Adam - hyperparameters : default diff --git a/enhancer/cli/train_config/dataset/Vctk.yaml b/enhancer/cli/train_config/dataset/Vctk.yaml index a1e1394..5c19320 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: 16 +batch_size: 64 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 2966c20..c2da54e 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 +metric : [mse,pesq,stoi] lr : 0.001 ReduceLr_patience : 10 ReduceLr_factor : 0.1