[common.engine.callbacks] Get lr from optimizer if no scheduler defined

src/mednet/libs/common/engine/callbacks.py

@@ -141,9 +141,13 @@ class LoggingCallback(lightning.pytorch.Callback):
         epoch_time = time.time() - self._start_training_epoch_time
 
         self._to_log["epoch-duration-seconds/train"] = epoch_time
-        self._to_log["learning-rate"] = pl_module.trainer.lr_scheduler_configs[
-            0
-        ].scheduler.optimizer.param_groups[0]["lr"]  # type: ignore
+
+        if len(pl_module.trainer.lr_scheduler_configs) == 0:
+            self._to_log["learning-rate"] = pl_module.optimizers().defaults["lr"]
+        else:
+            self._to_log["learning-rate"] = pl_module.trainer.lr_scheduler_configs[
+                0
+            ].scheduler.optimizer.param_groups[0]["lr"]  # type: ignore
 
         overall_cycle_time = time.time() - self._start_training_epoch_time
         self._to_log["cycle-time-seconds/train"] = overall_cycle_time