diff --git a/enhancer/cli/train_config/dataset/Vctk.yaml b/enhancer/cli/train_config/dataset/Vctk.yaml index 76a2006..62acea5 100644 --- a/enhancer/cli/train_config/dataset/Vctk.yaml +++ b/enhancer/cli/train_config/dataset/Vctk.yaml @@ -2,12 +2,13 @@ _target_: enhancer.data.dataset.EnhancerDataset name : vctk root_dir : /scratch/c.sistc3/DS_10283_2791 duration : 3 +stride : 0.5 sampling_rate: 16000 batch_size: 128 valid_minutes : 20 files: - train_clean : clean_trainset_56spk_wav + train_clean : clean_trainset_28spk_wav test_clean : clean_testset_wav - train_noisy : noisy_trainset_56spk_wav + train_noisy : noisy_trainset_28spk_wav test_noisy : noisy_testset_wav diff --git a/enhancer/cli/train_config/hyperparameters/default.yaml b/enhancer/cli/train_config/hyperparameters/default.yaml index fc99710..6410c0e 100644 --- a/enhancer/cli/train_config/hyperparameters/default.yaml +++ b/enhancer/cli/train_config/hyperparameters/default.yaml @@ -1,6 +1,6 @@ -loss : mse +loss : mae metric : [stoi,pesq,si-sdr] -lr : 0.001 +lr : 0.0003 ReduceLr_patience : 10 ReduceLr_factor : 0.5 min_lr : 0.00 diff --git a/enhancer/cli/train_config/trainer/default.yaml b/enhancer/cli/train_config/trainer/default.yaml index d4917e8..ff24f6e 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: 100 -max_epochs: 250 +max_epochs: 400 max_time: 00:47:00:00 min_epochs: 1 min_steps: null