summaryrefslogtreecommitdiff
path: root/text_recognizer/models
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2022-06-20 16:23:40 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2022-06-20 16:23:40 +0200
commitc1a536c5969391541f8dd096580d4a797c3b4d2c (patch)
tree0fbae656c138f1aee96fe7599b83613aa48529b1 /text_recognizer/models
parentefeca190eb62a7d0b470f02ae0fd07f0a3f11ef3 (diff)
Fix configuration of lr scheduler
Diffstat (limited to 'text_recognizer/models')
-rw-r--r--text_recognizer/models/base.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/text_recognizer/models/base.py b/text_recognizer/models/base.py
index 8dbdafa..70eb8dd 100644
--- a/text_recognizer/models/base.py
+++ b/text_recognizer/models/base.py
@@ -79,7 +79,7 @@ class LitBase(LightningModule):
"""Configures optimizer and lr scheduler."""
optimizer = self._configure_optimizer()
scheduler = self._configure_lr_schedulers(optimizer)
- return {"optimizer": optimizer, "scheduler": scheduler}
+ return {"optimizer": optimizer, "lr_scheduler": scheduler}
def forward(self, data: Tensor) -> Tensor:
"""Feedforward pass."""