From f7eb0a600c33a07dfa16dcfcf80b488d46ea4782 Mon Sep 17 00:00:00 2001 From: shahules786 Date: Fri, 14 Oct 2022 10:47:20 +0530 Subject: [PATCH] 500 epochs --- enhancer/cli/train_config/hyperparameters/default.yaml | 6 +++--- enhancer/cli/train_config/trainer/default.yaml | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/enhancer/cli/train_config/hyperparameters/default.yaml b/enhancer/cli/train_config/hyperparameters/default.yaml index b822ccc..56f0713 100644 --- a/enhancer/cli/train_config/hyperparameters/default.yaml +++ b/enhancer/cli/train_config/hyperparameters/default.yaml @@ -1,7 +1,7 @@ loss : mse metric : [stoi,pesq,si-sdr] lr : 0.001 -ReduceLr_patience : 5 -ReduceLr_factor : 0.1 -min_lr : 0.000001 +ReduceLr_patience : 10 +ReduceLr_factor : 0.5 +min_lr : 0.00 EarlyStopping_factor : 10 diff --git a/enhancer/cli/train_config/trainer/default.yaml b/enhancer/cli/train_config/trainer/default.yaml index 39932ba..beb497c 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: 100 -max_epochs: 100 +max_epochs: 500 max_time: null min_epochs: 1 min_steps: null