diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-08-30 23:01:00 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-08-30 23:01:00 +0200 |
| commit | 276c24bdc4817f2817b47b7a3a6bcfd9bb47b2ef (patch) | |
| tree | d7432951258e6473face3ad45bee04c5dad39235 /text_recognizer/models | |
| parent | ca7e91a2a97dc54b8201cc7d0b1ce3eb4a3ee1bd (diff) | |
Bug fix when loading no lr scheduler
Diffstat (limited to 'text_recognizer/models')
| -rw-r--r-- | text_recognizer/models/base.py | 2 |
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( |