From 73cc92505956c463f13648e3381055542b9ed975 Mon Sep 17 00:00:00 2001 From: shahules786 Date: Thu, 27 Oct 2022 21:26:56 +0530 Subject: [PATCH] fix earlystop --- enhancer/cli/train_config/hyperparameters/default.yaml | 4 ++-- enhancer/cli/train_config/trainer/default.yaml | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/enhancer/cli/train_config/hyperparameters/default.yaml b/enhancer/cli/train_config/hyperparameters/default.yaml index 63eeece..7a9d9a3 100644 --- a/enhancer/cli/train_config/hyperparameters/default.yaml +++ b/enhancer/cli/train_config/hyperparameters/default.yaml @@ -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 diff --git a/enhancer/cli/train_config/trainer/default.yaml b/enhancer/cli/train_config/trainer/default.yaml index 988c6e1..8bdf60f 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: 250 +max_epochs: 200 max_steps: -1 max_time: null min_epochs: 1