diff --git a/enhancer/cli/train_config/hyperparameters/default.yaml b/enhancer/cli/train_config/hyperparameters/default.yaml index a2f3042..c07a0df 100644 --- a/enhancer/cli/train_config/hyperparameters/default.yaml +++ b/enhancer/cli/train_config/hyperparameters/default.yaml @@ -1,7 +1,7 @@ loss : mse metric : [mse,stoi] lr : 0.001 -ReduceLr_patience : 10 +ReduceLr_patience : 5 ReduceLr_factor : 0.1 min_lr : 0.000001 -EarlyStopping_factor : 20 +EarlyStopping_factor : 10 diff --git a/enhancer/cli/train_config/trainer/default.yaml b/enhancer/cli/train_config/trainer/default.yaml index b64a7a6..b0ff02c 100644 --- a/enhancer/cli/train_config/trainer/default.yaml +++ b/enhancer/cli/train_config/trainer/default.yaml @@ -22,7 +22,7 @@ limit_predict_batches: 1.0 limit_test_batches: 1.0 limit_train_batches: 1.0 limit_val_batches: 1.0 -log_every_n_steps: 50 +log_every_n_steps: 100 max_epochs: 100 max_time: null min_epochs: 1