Seich
/
mayavoz
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

config

This commit is contained in:
shahules786 2022-11-23 19:26:16 +05:30
parent 530dab4648
commit 24ac484b23
4 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
enhancer/cli/train_config/config.yaml
View File

@ -1,5 +1,5 @@
defaults: defaults:
- model : Demucs - model : DCCRN
- dataset : DNS-2020 - dataset : DNS-2020
- optimizer : Adam - optimizer : Adam
- hyperparameters : default - hyperparameters : default

2
enhancer/cli/train_config/dataset/DNS-2020.yaml
View File

@ -1,7 +1,7 @@
_target_: enhancer.data.dataset.EnhancerDataset _target_: enhancer.data.dataset.EnhancerDataset
root_dir : /scratch/c.sistc3/MS-SNSD/DNS20 root_dir : /scratch/c.sistc3/MS-SNSD/DNS20
name : dns-2020 name : dns-2020
duration : 5 duration : 3
stride : 2 stride : 2
sampling_rate: 16000 sampling_rate: 16000
batch_size: 32 batch_size: 32

4
enhancer/cli/train_config/hyperparameters/default.yaml
View File

@ -1,6 +1,6 @@
loss : mae loss : si-snr
metric : [stoi,pesq] metric : [stoi,pesq]
lr : 0.0003 lr : 0.001
ReduceLr_patience : 10 ReduceLr_patience : 10
Early_stop : False Early_stop : False
ReduceLr_factor : 0.5 ReduceLr_factor : 0.5

2
enhancer/cli/train_config/mlflow/experiment.yaml
View File

@ -1,2 +1,2 @@
experiment_name : shahules/enhancer experiment_name : shahules/enhancer
run_name : demucs-dns20 run_name : dccrn-dns20