diff --git a/cli/train_config/dataset/Vctk.yaml b/cli/train_config/dataset/Vctk.yaml index aed10ac..129d9a8 100644 --- a/cli/train_config/dataset/Vctk.yaml +++ b/cli/train_config/dataset/Vctk.yaml @@ -6,9 +6,9 @@ sampling_rate: 16000 batch_size: 64 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/cli/train_config/hyperparameters/default.yaml b/cli/train_config/hyperparameters/default.yaml index 49a7f80..82ac3c2 100644 --- a/cli/train_config/hyperparameters/default.yaml +++ b/cli/train_config/hyperparameters/default.yaml @@ -1,7 +1,6 @@ loss : mse metric : mae lr : 0.0001 -num_epochs : 100 ReduceLr_patience : 5 ReduceLr_factor : 0.1 min_lr : 0.000001 diff --git a/cli/train_config/trainer/default.yaml b/cli/train_config/trainer/default.yaml index 2aa5083..55101de 100644 --- a/cli/train_config/trainer/default.yaml +++ b/cli/train_config/trainer/default.yaml @@ -22,8 +22,8 @@ limit_predict_batches: 1.0 limit_test_batches: 1.0 limit_train_batches: 1.0 limit_val_batches: 1.0 -log_every_n_steps: 10 -max_epochs: 30 +log_every_n_steps: 1 +max_epochs: 10 max_steps: null max_time: null min_epochs: 1