summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2022-06-19 21:03:11 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2022-06-19 21:03:11 +0200
commitce3f63801013aba2f05cfb92f1a3a87393610d27 (patch)
treef7e4d0a404f9db289336f830a08a975796c93134
parent8689e8da65ed02a13975956d70216d7f33db1688 (diff)
Fix variable names
-rw-r--r--text_recognizer/models/transformer.py6
1 files changed, 3 insertions, 3 deletions
diff --git a/text_recognizer/models/transformer.py b/text_recognizer/models/transformer.py
index 7afe9bd..7686c4f 100644
--- a/text_recognizer/models/transformer.py
+++ b/text_recognizer/models/transformer.py
@@ -17,8 +17,8 @@ class LitTransformer(LitBase):
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,
max_output_len: int = 451,
start_token: str = "<s>",
@@ -26,7 +26,7 @@ class LitTransformer(LitBase):
pad_token: str = "<p>",
) -> None:
super().__init__(
- network, loss_fn, optimizer_configs, lr_scheduler_configs, mapping
+ network, loss_fn, optimizer_config, lr_scheduler_config, mapping
)
self.max_output_len = max_output_len
self.start_token = start_token