diff --git a/cli/train_config/hyperparameters/default.yaml b/cli/train_config/hyperparameters/default.yaml index 5cbdcb0..4931c7c 100644 --- a/cli/train_config/hyperparameters/default.yaml +++ b/cli/train_config/hyperparameters/default.yaml @@ -1,4 +1,4 @@ loss : mse metric : mae lr : 0.001 -num_epochs : 10 +num_epochs : 100