diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-22 22:25:58 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-22 22:25:58 +0200 |
| commit | 801ff44c68c4208c75c78fe400a72d82ef494778 (patch) | |
| tree | 3e4d30df8e8f33269525dcd3509b3da61f4ed052 | |
| parent | 598749ee9df7a1d25d6f2240166e3441286e5517 (diff) | |
Add logging when loading optimizer
| -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 70eb8dd..64d3d71 100644 --- a/text_recognizer/models/base.py +++ b/text_recognizer/models/base.py @@ -49,6 +49,7 @@ class LitBase(LightningModule): def _configure_optimizer(self) -> Type[torch.optim.Optimizer]: """Configures the optimizer.""" + log.info(f"Instantiating optimizer <{self.optimizer_config._target_}>") return hydra.utils.instantiate( self.optimizer_config, params=self.network.parameters() ) |