From 8689e8da65ed02a13975956d70216d7f33db1688 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Sun, 19 Jun 2022 20:58:36 +0200 Subject: Fix lr and optim conf --- text_recognizer/models/base.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'text_recognizer') diff --git a/text_recognizer/models/base.py b/text_recognizer/models/base.py index 1ebb256..8dbdafa 100644 --- a/text_recognizer/models/base.py +++ b/text_recognizer/models/base.py @@ -20,16 +20,16 @@ class LitBase(LightningModule): self, network: Type[nn.Module], loss_fn: Type[nn.Module], - optimizer_configs: DictConfig, - lr_scheduler_configs: Optional[DictConfig], + optimizer_config: DictConfig, + lr_scheduler_config: Optional[DictConfig], mapping: EmnistMapping, ) -> None: super().__init__() self.network = network self.loss_fn = loss_fn - self.optimizer_configs = optimizer_configs - self.lr_scheduler_configs = lr_scheduler_configs + self.optimizer_config = optimizer_config + self.lr_scheduler_config = lr_scheduler_config self.mapping = mapping # Placeholders -- cgit v1.2.3-70-g09d2