config

2022-11-21 09:45:24 +05:30 · 2022-11-21 09:45:24 +05:30 · 9da80bbfb1
parent 75aa54b9d7
commit 9da80bbfb1
4 changed files with 5 additions and 4 deletions
--- a/enhancer/cli/train_config/config.yaml
+++ b/enhancer/cli/train_config/config.yaml
@ -1,5 +1,5 @@
 defaults:
-  - model : DCCRN
+  - model : Demucs
  - dataset : DNS-2020
  - optimizer : Adam
  - hyperparameters : default
--- a/enhancer/cli/train_config/dataset/DNS-2020.yaml
+++ b/enhancer/cli/train_config/dataset/DNS-2020.yaml
@ -1,7 +1,8 @@
 _target_: enhancer.data.dataset.EnhancerDataset
 root_dir : /scratch/c.sistc3/MS-SNSD/DNS15
 name : dns-2020
-duration : 2
+duration : 5
+stride : 2
 sampling_rate: 16000
 batch_size: 32
 min_valid_minutes: 25.0
--- 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.5
--- 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-DNS-15
+run_name : demucs-dns20