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 +