run waveunet

2022-10-11 16:51:41 +05:30 · 2022-10-11 16:51:41 +05:30 · 0c5db496e2
parent 031221b79e
commit 0c5db496e2
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/hyperparameters/default.yaml
+++ b/enhancer/cli/train_config/hyperparameters/default.yaml
@ -1,5 +1,5 @@
 loss : mse
-metric : [mse,pesq,stoi]
+metric : [mse,stoi]
 lr : 0.001
 ReduceLr_patience : 10
 ReduceLr_factor : 0.1
--- a/enhancer/cli/train_config/trainer/default.yaml
+++ b/enhancer/cli/train_config/trainer/default.yaml
@ -23,7 +23,7 @@ limit_test_batches: 1.0
 limit_train_batches: 1.0
 limit_val_batches: 1.0
 log_every_n_steps: 50
-max_epochs: 5
+max_epochs: 100
 max_time: null
 min_epochs: 1
 min_steps: null