From 9da80bbfb1c72f04f11ecef9f55af47ab390deeb Mon Sep 17 00:00:00 2001 From: shahules786 Date: Mon, 21 Nov 2022 09:45:24 +0530 Subject: [PATCH] config --- enhancer/cli/train_config/config.yaml | 2 +- enhancer/cli/train_config/dataset/DNS-2020.yaml | 3 ++- enhancer/cli/train_config/hyperparameters/default.yaml | 2 +- enhancer/cli/train_config/mlflow/experiment.yaml | 2 +- 4 files changed, 5 insertions(+), 4 deletions(-) diff --git a/enhancer/cli/train_config/config.yaml b/enhancer/cli/train_config/config.yaml index aea9269..9c464ac 100644 --- a/enhancer/cli/train_config/config.yaml +++ b/enhancer/cli/train_config/config.yaml @@ -1,5 +1,5 @@ defaults: - - model : DCCRN + - model : Demucs - dataset : DNS-2020 - optimizer : Adam - hyperparameters : default diff --git a/enhancer/cli/train_config/dataset/DNS-2020.yaml b/enhancer/cli/train_config/dataset/DNS-2020.yaml index 09e56a3..c2e13f9 100644 --- a/enhancer/cli/train_config/dataset/DNS-2020.yaml +++ b/enhancer/cli/train_config/dataset/DNS-2020.yaml @@ -1,7 +1,8 @@ _target_: enhancer.data.dataset.EnhancerDataset root_dir : /scratch/c.sistc3/MS-SNSD/DNS15 name : dns-2020 -duration : 2 +duration : 5 +stride : 2 sampling_rate: 16000 batch_size: 32 min_valid_minutes: 25.0 diff --git a/enhancer/cli/train_config/hyperparameters/default.yaml b/enhancer/cli/train_config/hyperparameters/default.yaml index 93db6aa..af40c41 100644 --- a/enhancer/cli/train_config/hyperparameters/default.yaml +++ b/enhancer/cli/train_config/hyperparameters/default.yaml @@ -1,6 +1,6 @@ loss : si-snr metric : [stoi,pesq] -lr : 0.001 +lr : 0.0003 ReduceLr_patience : 10 Early_stop : False ReduceLr_factor : 0.5 diff --git a/enhancer/cli/train_config/mlflow/experiment.yaml b/enhancer/cli/train_config/mlflow/experiment.yaml index 36af7ed..840b6dc 100644 --- a/enhancer/cli/train_config/mlflow/experiment.yaml +++ b/enhancer/cli/train_config/mlflow/experiment.yaml @@ -1,2 +1,2 @@ experiment_name : shahules/enhancer -run_name : DCCRN-DNS-15 +run_name : demucs-dns20