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

fix earlystop

This commit is contained in:
shahules786 2022-10-27 21:26:56 +05:30
parent 3a9e577ccb
commit 73cc925059
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,8 +1,8 @@
loss : mae
metric : [stoi,pesq,si-sdr]
lr : 0.0003
ReduceLr_patience : 10
ReduceLr_patience : 8
Early_stop : True
ReduceLr_factor : 0.5
min_lr : 0.000001
EarlyStopping_patience : 10
EarlyStopping_patience : 15

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_val_batches: 1.0
log_every_n_steps: 50
max_epochs: 250
max_epochs: 200
max_steps: -1
max_time: null
min_epochs: 1