diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-04-05 23:25:25 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-04-05 23:25:25 +0200 |
| commit | 5e11924ca6aaea7898caca94675f41f67706a406 (patch) | |
| tree | ab1e3d9ea4a8052ecbe6625821720c27e067d2f3 | |
| parent | dedf8deb025ac9efdad5e9baf9165ef63d6829ff (diff) | |
Fix typing
| -rw-r--r-- | text_recognizer/models/base.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/text_recognizer/models/base.py b/text_recognizer/models/base.py index 11d1eb1..f4df3f3 100644 --- a/text_recognizer/models/base.py +++ b/text_recognizer/models/base.py @@ -40,7 +40,7 @@ class LitBaseModel(pl.LightningModule): args = {} or criterion.args return getattr(nn, criterion.type)(**args) - def _configure_optimizer(self) -> torch.optim.Optimizer: + def _configure_optimizer(self) -> Type[torch.optim.Optimizer]: """Configures the optimizer.""" args = {} or self._optimizer.args if self._optimizer.type == "MADGRAD": |