testrun for metrics

2022-10-10 12:49:41 +05:30 · 2022-10-10 12:49:41 +05:30 · a417e226f3
parent 5d8f49d78e
commit a417e226f3
3 changed files with 3 additions and 3 deletions
--- a/enhancer/cli/train_config/config.yaml
+++ b/enhancer/cli/train_config/config.yaml
@ -1,5 +1,5 @@
 defaults:
-  - model : Demucs
+  - model : WaveUnet
  - dataset : Vctk
  - optimizer : Adam
  - hyperparameters : default
--- a/enhancer/cli/train_config/dataset/Vctk.yaml
+++ b/enhancer/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: 16
+batch_size: 64

 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
+metric : [mse,pesq,stoi]
 lr : 0.001
 ReduceLr_patience : 10
 ReduceLr_factor : 0.1