Fix a wrong HPO log (#3972)

* fix wrong log * fix wrong hpo log
openvinotoolkit · Sep 23, 2024 · 0e2ab6c · 0e2ab6c
1 parent 40f5ddd
commit 0e2ab6c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/otx/engine/hpo/hpo_api.py b/src/otx/engine/hpo/hpo_api.py
@@ -336,8 +336,11 @@ def _adapt_batch_size_search_space(self, adapt_mode: Literal["Safe", "Full"]) ->
 
         self._engine.datamodule.train_subset.batch_size = origin_bs
         self._engine.model.optimizer_callable.optimizer_kwargs["lr"] = origin_lr  # type: ignore[attr-defined]
+        logger.info(
+            "Max value of batch size search space : "
+            f"{self._hpo_config['search_space']['datamodule.train_subset.batch_size']['max']} -> {adapted_bs}",
+        )
         self._hpo_config["search_space"]["datamodule.train_subset.batch_size"]["max"] = adapted_bs
-        logger.info(f"Max value of batch size search space : {origin_bs} -> {adapted_bs}")
 
     @staticmethod
     def _remove_wrong_search_space(search_space: dict[str, dict[str, Any]]) -> None: