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