From 24ac484b236c0d5c5f3875c761601dc105bc463f Mon Sep 17 00:00:00 2001 From: shahules786 Date: Wed, 23 Nov 2022 19:26:16 +0530 Subject: [PATCH] config --- enhancer/cli/train_config/config.yaml | 2 +- enhancer/cli/train_config/dataset/DNS-2020.yaml | 2 +- enhancer/cli/train_config/hyperparameters/default.yaml | 4 ++-- enhancer/cli/train_config/mlflow/experiment.yaml | 2 +- 4 files changed, 5 insertions(+), 5 deletions(-) diff --git a/enhancer/cli/train_config/config.yaml b/enhancer/cli/train_config/config.yaml index 9c464ac..aea9269 100644 --- a/enhancer/cli/train_config/config.yaml +++ b/enhancer/cli/train_config/config.yaml @@ -1,5 +1,5 @@ defaults: - - model : Demucs + - model : DCCRN - 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 530b49a..6216103 100644 --- a/enhancer/cli/train_config/dataset/DNS-2020.yaml +++ b/enhancer/cli/train_config/dataset/DNS-2020.yaml @@ -1,7 +1,7 @@ _target_: enhancer.data.dataset.EnhancerDataset root_dir : /scratch/c.sistc3/MS-SNSD/DNS20 name : dns-2020 -duration : 5 +duration : 3 stride : 2 sampling_rate: 16000 batch_size: 32 diff --git a/enhancer/cli/train_config/hyperparameters/default.yaml b/enhancer/cli/train_config/hyperparameters/default.yaml index 9b9bd7b..93db6aa 100644 --- a/enhancer/cli/train_config/hyperparameters/default.yaml +++ b/enhancer/cli/train_config/hyperparameters/default.yaml @@ -1,6 +1,6 @@ -loss : mae +loss : si-snr metric : [stoi,pesq] -lr : 0.0003 +lr : 0.001 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 840b6dc..bbe8180 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 : demucs-dns20 +run_name : dccrn-dns20