From 8517fbecb6a039e185284a24c65dc953838d44f0 Mon Sep 17 00:00:00 2001 From: Tiziano Date: Mon, 2 Mar 2026 11:31:48 +0100 Subject: [PATCH] fix: set pytorch lightning log level to ERROR --- hyperbench/train/__init__.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/hyperbench/train/__init__.py b/hyperbench/train/__init__.py index f18d0e6..d80cf77 100644 --- a/hyperbench/train/__init__.py +++ b/hyperbench/train/__init__.py @@ -1,3 +1,8 @@ +import logging + +logging.getLogger("lightning.pytorch").setLevel(logging.ERROR) + + from .negative_sampler import NegativeSampler, RandomNegativeSampler from .negative_sampling_scheduler import NegativeSamplingSchedule, NegativeSamplingScheduler from .trainer import MultiModelTrainer