summaryrefslogtreecommitdiff
path: root/text_recognizer/models
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2022-06-22 22:25:58 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2022-06-22 22:25:58 +0200
commit801ff44c68c4208c75c78fe400a72d82ef494778 (patch)
tree3e4d30df8e8f33269525dcd3509b3da61f4ed052 /text_recognizer/models
parent598749ee9df7a1d25d6f2240166e3441286e5517 (diff)
Add logging when loading optimizer
Diffstat (limited to 'text_recognizer/models')
-rw-r--r--text_recognizer/models/base.py1
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()
)