diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-19 21:03:11 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-19 21:03:11 +0200 |
| commit | ce3f63801013aba2f05cfb92f1a3a87393610d27 (patch) | |
| tree | f7e4d0a404f9db289336f830a08a975796c93134 /text_recognizer | |
| parent | 8689e8da65ed02a13975956d70216d7f33db1688 (diff) | |
Fix variable names
Diffstat (limited to 'text_recognizer')
| -rw-r--r-- | text_recognizer/models/transformer.py | 6 |
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 |