Set verbosity for optuna

aimclub · Aug 4, 2023 · eb7eccc · eb7eccc
1 parent aec8c8d
commit eb7eccc
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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,