config

2022-11-07 19:38:27 +05:30 · 2022-11-07 19:38:27 +05:30 · c4e392aff5
parent 24fa16ca25
commit c4e392aff5
5 changed files with 6 additions and 6 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/dataset/DNS-2020.yaml
+++ b/enhancer/cli/train_config/dataset/DNS-2020.yaml
@ -1,5 +1,5 @@
 _target_: enhancer.data.dataset.EnhancerDataset
-root_dir : /scratch/c.sistc3/MS-SNSD/DNS15
+root_dir : /scratch/c.sistc3/MS-SNSD/DNS30
 name : dns-2020
 duration : 5.0
 stride : 1
--- a/enhancer/cli/train_config/hyperparameters/default.yaml
+++ b/enhancer/cli/train_config/hyperparameters/default.yaml
@ -1,6 +1,6 @@
 loss : si-snr
 metric : [stoi,pesq]
-lr : 0.001
+lr : 0.0003
 ReduceLr_patience : 10
 Early_stop : False
 ReduceLr_factor : 0.2
--- 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 : Demucs-DNS
+run_name : Demucs-DNS30
--- a/enhancer/cli/train_config/trainer/default.yaml
+++ b/enhancer/cli/train_config/trainer/default.yaml
@ -25,7 +25,7 @@ limit_val_batches: 1.0
 log_every_n_steps: 50
 max_epochs: 100
 max_steps: -1
-max_time: 00:47:00:00
+max_time: null
 min_epochs: 1
 min_steps: null
 move_metrics_to_cpu: False