run 100 epochs

2022-10-13 10:52:22 +05:30 · 2022-10-13 10:52:22 +05:30 · 8a55a77640
parent 94a4ea38ed
commit 8a55a77640
2 changed files with 2 additions and 2 deletions
--- a/enhancer/cli/train_config/dataset/Vctk.yaml
+++ b/enhancer/cli/train_config/dataset/Vctk.yaml
@ -4,7 +4,7 @@ root_dir : /scratch/c.sistc3/DS_10283_2791
 duration : 1.0
 sampling_rate: 16000
 batch_size: 128
-valid_size : 0.1
+valid_size : 0.05

 files:
  train_clean : clean_trainset_28spk_wav
--- a/enhancer/cli/train_config/hyperparameters/default.yaml
+++ b/enhancer/cli/train_config/hyperparameters/default.yaml
@ -1,5 +1,5 @@
 loss : mse
-metric : [mse,stoi,pesq]
+metric : [stoi,pesq,si-sdr]
 lr : 0.001
 ReduceLr_patience : 5
 ReduceLr_factor : 0.1