config

2022-11-09 19:07:53 +05:30 · 2022-11-09 19:07:53 +05:30 · b5582832f3
parent 86b71ce090
commit b5582832f3
4 changed files with 7 additions and 7 deletions
--- a/enhancer/cli/train_config/config.yaml
+++ b/enhancer/cli/train_config/config.yaml
@ -1,6 +1,6 @@
 defaults:
-  - model : DCCRN
-  - dataset : Vctk
+  - model : Demucs
+  - dataset : DNS-2020
  - optimizer : Adam
  - hyperparameters : default
  - trainer : default
--- a/enhancer/cli/train_config/hyperparameters/default.yaml
+++ b/enhancer/cli/train_config/hyperparameters/default.yaml
@ -1,8 +1,8 @@
 loss : si-snr
 metric : [stoi,pesq]
-lr : 0.001
-ReduceLr_patience : 10
+lr : 0.0003
+ReduceLr_patience : 100
 Early_stop : False
-ReduceLr_factor : 0.2
+ReduceLr_factor : 0.1
 min_lr : 0.0000001
 EarlyStopping_patience : 10
--- a/enhancer/cli/train_config/mlflow/experiment.yaml
+++ b/enhancer/cli/train_config/mlflow/experiment.yaml
@ -1,2 +1,2 @@
 experiment_name : shahules/enhancer
-run_name : DCCRN-Vctk28
+run_name : Demucs-DNS-2020(30)
--- 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: 100
+max_epochs: 150
 max_steps: -1
 max_time: null
 min_epochs: 1