From c4e392aff572dfcea3735c64a23123028637adfa Mon Sep 17 00:00:00 2001 From: shahules786 Date: Mon, 7 Nov 2022 19:38:27 +0530 Subject: [PATCH] config --- enhancer/cli/train_config/config.yaml | 4 ++-- enhancer/cli/train_config/dataset/DNS-2020.yaml | 2 +- enhancer/cli/train_config/hyperparameters/default.yaml | 2 +- enhancer/cli/train_config/mlflow/experiment.yaml | 2 +- enhancer/cli/train_config/trainer/default.yaml | 2 +- 5 files changed, 6 insertions(+), 6 deletions(-) diff --git a/enhancer/cli/train_config/config.yaml b/enhancer/cli/train_config/config.yaml index 77877c6..9c464ac 100644 --- a/enhancer/cli/train_config/config.yaml +++ b/enhancer/cli/train_config/config.yaml @@ -1,6 +1,6 @@ defaults: - - model : DCCRN - - dataset : Vctk + - model : Demucs + - dataset : DNS-2020 - optimizer : Adam - hyperparameters : default - trainer : default diff --git a/enhancer/cli/train_config/dataset/DNS-2020.yaml b/enhancer/cli/train_config/dataset/DNS-2020.yaml index 159ecf7..3a1976b 100644 --- a/enhancer/cli/train_config/dataset/DNS-2020.yaml +++ b/enhancer/cli/train_config/dataset/DNS-2020.yaml @@ -1,5 +1,5 @@ _target_: enhancer.data.dataset.EnhancerDataset -root_dir : /scratch/c.sistc3/MS-SNSD/DNS15 +root_dir : /scratch/c.sistc3/MS-SNSD/DNS30 name : dns-2020 duration : 5.0 stride : 1 diff --git a/enhancer/cli/train_config/hyperparameters/default.yaml b/enhancer/cli/train_config/hyperparameters/default.yaml index 80fb033..356181c 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.2 diff --git a/enhancer/cli/train_config/mlflow/experiment.yaml b/enhancer/cli/train_config/mlflow/experiment.yaml index b907f62..87e7895 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-DNS +run_name : Demucs-DNS30 diff --git a/enhancer/cli/train_config/trainer/default.yaml b/enhancer/cli/train_config/trainer/default.yaml index 06f7fd0..b23133f 100644 --- a/enhancer/cli/train_config/trainer/default.yaml +++ b/enhancer/cli/train_config/trainer/default.yaml @@ -25,7 +25,7 @@ limit_val_batches: 1.0 log_every_n_steps: 50 max_epochs: 100 max_steps: -1 -max_time: 00:47:00:00 +max_time: null min_epochs: 1 min_steps: null move_metrics_to_cpu: False