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