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

run waveunet

This commit is contained in:
shahules786 2022-10-11 16:51:41 +05:30
parent 031221b79e
commit 0c5db496e2
3 changed files with 3 additions and 3 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 : WaveUnet
- dataset : Vctk - dataset : Vctk
- optimizer : Adam - optimizer : Adam
- hyperparameters : default - hyperparameters : default

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

@ -1,5 +1,5 @@
loss : mse loss : mse
metric : [mse,pesq,stoi] metric : [mse,stoi]
lr : 0.001 lr : 0.001
ReduceLr_patience : 10 ReduceLr_patience : 10
ReduceLr_factor : 0.1 ReduceLr_factor : 0.1

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: 5 max_epochs: 100
max_time: null max_time: null
min_epochs: 1 min_epochs: 1
min_steps: null min_steps: null