From 801ff44c68c4208c75c78fe400a72d82ef494778 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Wed, 22 Jun 2022 22:25:58 +0200 Subject: Add logging when loading optimizer --- text_recognizer/models/base.py | 1 + 1 file changed, 1 insertion(+) (limited to 'text_recognizer') 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() ) -- cgit v1.2.3-70-g09d2