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

[libs.common.engine.callbacks] Fix access to last learning-rate

parent df66e72e
No related branches found
No related tags found
1 merge request!46Create common library
Pipeline #89293 passed
......@@ -147,7 +147,7 @@ class LoggingCallback(lightning.pytorch.Callback):
else:
self._to_log["learning-rate"] = pl_module.trainer.lr_scheduler_configs[
0
].scheduler.get_last_lr() # type: ignore
].scheduler.get_last_lr()[0] # type: ignore
overall_cycle_time = time.time() - self._start_training_epoch_time
self._to_log["cycle-time-seconds/train"] = overall_cycle_time
......
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