Skip to content
Snippets Groups Projects
Commit 6f2f47f6 authored by André Anjos's avatar André Anjos :speech_balloon:
Browse files

[libs.common.engine.callbacks] Use appropriate interface for extracting the...

[libs.common.engine.callbacks] Use appropriate interface for extracting the current learning rate from schedulers
parent d0a13cb3
No related branches found
No related tags found
1 merge request!46Create common library
Pipeline #89290 failed
...@@ -145,6 +145,7 @@ class LoggingCallback(lightning.pytorch.Callback): ...@@ -145,6 +145,7 @@ class LoggingCallback(lightning.pytorch.Callback):
if len(pl_module.trainer.lr_scheduler_configs) == 0: 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"]
else: else:
__import__("pdb").set_trace()
self._to_log["learning-rate"] = pl_module.trainer.lr_scheduler_configs[ self._to_log["learning-rate"] = pl_module.trainer.lr_scheduler_configs[
0 0
].scheduler.optimizer.param_groups[0]["lr"] # type: ignore ].scheduler.optimizer.param_groups[0]["lr"] # type: ignore
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment