Bug fix when loading no lr scheduler

1 files changed, 1 insertions, 1 deletions

diff --git a/text_recognizer/models/base.py b/text_recognizer/models/base.py
index 56d4ca5..57d319e 100644
--- a/text_recognizer/models/base.py
+++ b/text_recognizer/models/base.py
@@ -63,7 +63,7 @@ class BaseLitModel(LightningModule):
         self, optimizers: List[Type[torch.optim.Optimizer]]
     ) -> List[Dict[str, Any]]:
         """Configures the lr scheduler."""
-        if None in self.lr_scheduler_configs:
+        if self.lr_scheduler_configs is None:
             return []
         schedulers = []
         for optimizer, lr_scheduler_config in zip(