diff options
-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() ) |