From 6faffdcb1731b9b0f799a68afaf3f87baba44794 Mon Sep 17 00:00:00 2001 From: shahules786 Date: Sun, 2 Oct 2022 18:18:56 +0530 Subject: [PATCH] include more params --- cli/train_config/hyperparameters/default.yaml | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/cli/train_config/hyperparameters/default.yaml b/cli/train_config/hyperparameters/default.yaml index 04b099b..49a7f80 100644 --- a/cli/train_config/hyperparameters/default.yaml +++ b/cli/train_config/hyperparameters/default.yaml @@ -2,3 +2,8 @@ loss : mse metric : mae lr : 0.0001 num_epochs : 100 +ReduceLr_patience : 5 +ReduceLr_factor : 0.1 +min_lr : 0.000001 +EarlyStopping_factor : 10 +