From c1a536c5969391541f8dd096580d4a797c3b4d2c Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Mon, 20 Jun 2022 16:23:40 +0200 Subject: Fix configuration of lr scheduler --- text_recognizer/models/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'text_recognizer/models') 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.""" -- cgit v1.2.3-70-g09d2