From 1dcaa7fbe05313487c34fc68c28640e4cc130036 Mon Sep 17 00:00:00 2001 From: shahules786 Date: Sun, 2 Oct 2022 09:05:02 +0530 Subject: [PATCH] change config --- cli/train_config/config.yaml | 2 +- cli/train_config/dataset/Vctk.yaml | 2 +- cli/train_config/hyperparameters/default.yaml | 1 - cli/train_config/trainer/default.yaml | 4 ++-- 4 files changed, 4 insertions(+), 5 deletions(-) diff --git a/cli/train_config/config.yaml b/cli/train_config/config.yaml index 6b5d98e..61551bd 100644 --- a/cli/train_config/config.yaml +++ b/cli/train_config/config.yaml @@ -1,5 +1,5 @@ defaults: - - model : Demucs + - model : WaveUnet - dataset : Vctk - optimizer : Adam - hyperparameters : default diff --git a/cli/train_config/dataset/Vctk.yaml b/cli/train_config/dataset/Vctk.yaml index d1c8646..aed10ac 100644 --- a/cli/train_config/dataset/Vctk.yaml +++ b/cli/train_config/dataset/Vctk.yaml @@ -3,7 +3,7 @@ name : vctk root_dir : /scratch/c.sistc3/DS_10283_2791 duration : 1.0 sampling_rate: 16000 -batch_size: 8 +batch_size: 64 files: train_clean : clean_trainset_56spk_wav diff --git a/cli/train_config/hyperparameters/default.yaml b/cli/train_config/hyperparameters/default.yaml index 04b099b..a0ac704 100644 --- a/cli/train_config/hyperparameters/default.yaml +++ b/cli/train_config/hyperparameters/default.yaml @@ -1,4 +1,3 @@ loss : mse metric : mae lr : 0.0001 -num_epochs : 100 diff --git a/cli/train_config/trainer/default.yaml b/cli/train_config/trainer/default.yaml index ab4e273..6c693d8 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: 100 +log_every_n_steps: 50 +max_epochs: 500 max_steps: null max_time: null min_epochs: 1