From a869dbb68778efc58ea8bf349808d78fc605b9e5 Mon Sep 17 00:00:00 2001 From: Onuralp SEZER Date: Tue, 4 Feb 2025 22:31:13 +0300 Subject: [PATCH] =?UTF-8?q?fix(logging):=20=F0=9F=90=9E=20correct=20logger?= =?UTF-8?q?=20variable=20names=20and=20set=20logging=20levels=20for=20clar?= =?UTF-8?q?ity?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Onuralp SEZER --- maestro/trainer/logger.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/maestro/trainer/logger.py b/maestro/trainer/logger.py index f2bd205..0520b59 100644 --- a/maestro/trainer/logger.py +++ b/maestro/trainer/logger.py @@ -37,13 +37,16 @@ def set_lightning_logging(level: str) -> None: from maestro.trainer import set_lightning_logging set_lightning_logging("DEBUG") """ + + lightning_logging = logging.getLogger("lightning") pytorch_lightning_logging = logging.getLogger("pytorch_lightning") - cuda_log = logging.getLogger("lightning.pytorch.accelerators.cuda") - rank_zero = logging.getLogger("lightning.pytorch.utilities.rank_zero") + cuda_logging = logging.getLogger("lightning.pytorch.accelerators.cuda") + rank_zero_logging = logging.getLogger("lightning.pytorch.utilities.rank_zero") + lightning_logging.setLevel(getattr(logging, level)) pytorch_lightning_logging.setLevel(getattr(logging, level)) - cuda_log.setLevel(getattr(logging, level)) - rank_zero.setLevel(getattr(logging, level)) + cuda_logging.setLevel(getattr(logging, level)) + rank_zero_logging.setLevel(getattr(logging, level)) def set_transformers_progress(status: bool) -> None: