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

config

This commit is contained in:
shahules786 2022-11-10 10:41:07 +05:30
parent effb4b03fb
commit ca6797c3f1
4 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

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

2
enhancer/cli/train_config/dataset/Vctk.yaml
View File

@ -4,7 +4,7 @@ root_dir : /scratch/c.sistc3/DS_10283_2791
duration : 2 duration : 2
stride : 1 stride : 1
sampling_rate: 16000 sampling_rate: 16000
batch_size: 128 batch_size: 64
min_valid_minutes : 25 min_valid_minutes : 25
files: files:
train_clean : clean_trainset_28spk_wav train_clean : clean_trainset_28spk_wav

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

@ -1,7 +1,7 @@
loss : mae loss : si-snr
metric : [si-sdr,stoi,pesq] metric : [stoi,pesq]
lr : 0.0003 lr : 0.001
ReduceLr_patience : 100 ReduceLr_patience : 10
Early_stop : False Early_stop : False
ReduceLr_factor : 0.1 ReduceLr_factor : 0.1
min_lr : 0.0000001 min_lr : 0.0000001

2
enhancer/cli/train_config/trainer/default.yaml
View File

@ -23,7 +23,7 @@ limit_test_batches: 1.0
limit_train_batches: 1.0 limit_train_batches: 1.0
limit_val_batches: 1.0 limit_val_batches: 1.0
log_every_n_steps: 50 log_every_n_steps: 50
max_epochs: 150 max_epochs: 100
max_steps: -1 max_steps: -1
max_time: null max_time: null
min_epochs: 1 min_epochs: 1