diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-04-11 21:48:34 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-04-11 21:48:34 +0200 |
| commit | 0ab820d3595e4f17d4f1f3c310e186692f65cc67 (patch) | |
| tree | 21891ab98c10e64ef9261c69b2d494f42cda66f1 /text_recognizer/models/base.py | |
| parent | a548e421314908771ce9e413d9fa4e205943cceb (diff) | |
Working on mapping
Diffstat (limited to 'text_recognizer/models/base.py')
| -rw-r--r-- | text_recognizer/models/base.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/text_recognizer/models/base.py b/text_recognizer/models/base.py index 0928e6c..c6d5d73 100644 --- a/text_recognizer/models/base.py +++ b/text_recognizer/models/base.py @@ -60,6 +60,7 @@ class LitBaseModel(pl.LightningModule): scheduler["scheduler"] = getattr( torch.optim.lr_scheduler, self._lr_scheduler.type )(optimizer, **args) + return scheduler def configure_optimizers(self) -> Tuple[List[type], List[Dict[str, Any]]]: |