diff --git a/enhancer/cli/train_config/config.yaml b/enhancer/cli/train_config/config.yaml index 9c464ac..77877c6 100644 --- a/enhancer/cli/train_config/config.yaml +++ b/enhancer/cli/train_config/config.yaml @@ -1,6 +1,6 @@ defaults: - - model : Demucs - - dataset : DNS-2020 + - model : DCCRN + - dataset : Vctk - optimizer : Adam - hyperparameters : default - trainer : default diff --git a/enhancer/cli/train_config/dataset/Vctk.yaml b/enhancer/cli/train_config/dataset/Vctk.yaml index 208098a..ecfbb85 100644 --- a/enhancer/cli/train_config/dataset/Vctk.yaml +++ b/enhancer/cli/train_config/dataset/Vctk.yaml @@ -4,7 +4,7 @@ root_dir : /scratch/c.sistc3/DS_10283_2791 duration : 2 stride : 1 sampling_rate: 16000 -batch_size: 128 +batch_size: 64 min_valid_minutes : 25 files: train_clean : clean_trainset_28spk_wav diff --git a/enhancer/cli/train_config/hyperparameters/default.yaml b/enhancer/cli/train_config/hyperparameters/default.yaml index 0392f55..0bcb9e0 100644 --- a/enhancer/cli/train_config/hyperparameters/default.yaml +++ b/enhancer/cli/train_config/hyperparameters/default.yaml @@ -1,7 +1,7 @@ -loss : mae -metric : [si-sdr,stoi,pesq] -lr : 0.0003 -ReduceLr_patience : 100 +loss : si-snr +metric : [stoi,pesq] +lr : 0.001 +ReduceLr_patience : 10 Early_stop : False ReduceLr_factor : 0.1 min_lr : 0.0000001 diff --git a/enhancer/cli/train_config/trainer/default.yaml b/enhancer/cli/train_config/trainer/default.yaml index 797d259..b23133f 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: 150 +max_epochs: 100 max_steps: -1 max_time: null min_epochs: 1