From eb7eccc2171b7f1d2a55be49db153b67a0ddc714 Mon Sep 17 00:00:00 2001 From: Lyubov Yamshchikova Date: Fri, 4 Aug 2023 13:37:24 +0300 Subject: [PATCH] Set verbosity for optuna --- golem/core/tuning/optuna_tuner.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/golem/core/tuning/optuna_tuner.py b/golem/core/tuning/optuna_tuner.py index f592e2e6..aabcb95d 100644 --- a/golem/core/tuning/optuna_tuner.py +++ b/golem/core/tuning/optuna_tuner.py @@ -54,6 +54,9 @@ def tune(self, graph: DomainGraphForTune, show_progress: bool = True) -> \ if init_parameters: self.study.enqueue_trial(init_parameters) + verbosity_level = optuna.logging.INFO if show_progress else optuna.logging.WARNING + optuna.logging.set_verbosity(verbosity_level) + self.study.optimize(predefined_objective, n_trials=self.iterations, n_jobs=self.n_jobs,