diff --git a/enhancer/cli/train_config/hyperparameters/default.yaml b/enhancer/cli/train_config/hyperparameters/default.yaml index 09d3ebb..929905a 100644 --- a/enhancer/cli/train_config/hyperparameters/default.yaml +++ b/enhancer/cli/train_config/hyperparameters/default.yaml @@ -3,6 +3,6 @@ metric : [stoi,pesq,si-sdr] lr : 0.0003 ReduceLr_patience : 5 Early_stop : True -ReduceLr_factor : 0.25 +ReduceLr_factor : 0.1 min_lr : 0.0000001 -EarlyStopping_patience : 15 +EarlyStopping_patience : 10 diff --git a/enhancer/cli/train_config/trainer/default.yaml b/enhancer/cli/train_config/trainer/default.yaml index bd11ad7..b6c4441 100644 --- a/enhancer/cli/train_config/trainer/default.yaml +++ b/enhancer/cli/train_config/trainer/default.yaml @@ -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: 3 +max_epochs: 200 max_steps: -1 max_time: null min_epochs: 1