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/hyperparameters/default.yaml b/enhancer/cli/train_config/hyperparameters/default.yaml index 93db6aa..9b9bd7b 100644 --- a/enhancer/cli/train_config/hyperparameters/default.yaml +++ b/enhancer/cli/train_config/hyperparameters/default.yaml @@ -1,6 +1,6 @@ -loss : si-snr +loss : mae 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 393ad91..b907f62 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-VCTK +run_name : Demucs-DNS diff --git a/enhancer/cli/train_config/trainer/default.yaml b/enhancer/cli/train_config/trainer/default.yaml index b23133f..797d259 100644 --- a/enhancer/cli/train_config/trainer/default.yaml +++ b/enhancer/cli/train_config/trainer/default.yaml @@ -23,7 +23,7 @@ limit_test_batches: 1.0 limit_train_batches: 1.0 limit_val_batches: 1.0 log_every_n_steps: 50 -max_epochs: 100 +max_epochs: 150 max_steps: -1 max_time: null min_epochs: 1