diff --git a/enhancer/cli/train_config/dataset/Vctk.yaml b/enhancer/cli/train_config/dataset/Vctk.yaml index 2f22146..3f8def6 100644 --- a/enhancer/cli/train_config/dataset/Vctk.yaml +++ b/enhancer/cli/train_config/dataset/Vctk.yaml @@ -1,9 +1,9 @@ _target_: enhancer.data.dataset.EnhancerDataset name : vctk root_dir : /scratch/c.sistc3/DS_10283_2791 -duration : 2 +duration : 4.5 sampling_rate: 16000 -batch_size: 128 +batch_size: 32 valid_minutes : 15 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 0291c8e..b6bba46 100644 --- a/enhancer/cli/train_config/hyperparameters/default.yaml +++ b/enhancer/cli/train_config/hyperparameters/default.yaml @@ -1,7 +1,7 @@ loss : mse metric : [stoi,pesq,si-sdr] -lr : 0.001 -ReduceLr_patience : 10 -ReduceLr_factor : 0.5 +lr : 0.0001 +ReduceLr_patience : 5 +ReduceLr_factor : 0.2 min_lr : 0.000001 EarlyStopping_factor : 10