From e70da2f16368b9d6327071016f740b9ce927235f Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.dos.anjos@gmail.com>
Date: Tue, 2 Jul 2024 14:49:09 +0200
Subject: [PATCH] [libs.common.engine.callbacks] Remove call to pdb
---
src/mednet/libs/common/engine/callbacks.py | 5 ++---
1 file changed, 2 insertions(+), 3 deletions(-)
diff --git a/src/mednet/libs/common/engine/callbacks.py b/src/mednet/libs/common/engine/callbacks.py
index 4ef56073..0f17f098 100644
--- a/src/mednet/libs/common/engine/callbacks.py
+++ b/src/mednet/libs/common/engine/callbacks.py
@@ -143,12 +143,11 @@ class LoggingCallback(lightning.pytorch.Callback):
self._to_log["epoch-duration-seconds/train"] = epoch_time
if len(pl_module.trainer.lr_scheduler_configs) == 0:
- self._to_log["learning-rate"] = pl_module.optimizers().defaults["lr"]
+ self._to_log["learning-rate"] = pl_module.optimizers().defaults["lr"] # type: ignore
else:
- __import__("pdb").set_trace()
self._to_log["learning-rate"] = pl_module.trainer.lr_scheduler_configs[
0
- ].scheduler.optimizer.param_groups[0]["lr"] # type: ignore
+ ].scheduler.get_last_lr() # type: ignore
overall_cycle_time = time.time() - self._start_training_epoch_time
self._to_log["cycle-time-seconds/train"] = overall_cycle_time
--
GitLab