change config

2022-10-02 09:05:02 +05:30 · 2022-10-02 09:05:02 +05:30 · 1dcaa7fbe0
parent d25be3a59d
commit 1dcaa7fbe0
4 changed files with 4 additions and 5 deletions
--- 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
--- 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
--- 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
--- 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