From 7ed6145b970d58ebc05f8018c0b59fdb54ba6882 Mon Sep 17 00:00:00 2001
From: dcarron <daniel.carron@idiap.ch>
Date: Tue, 11 Apr 2023 10:14:16 +0200
Subject: [PATCH] Fixed loading of learning rate
---
src/ptbench/engine/callbacks.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/ptbench/engine/callbacks.py b/src/ptbench/engine/callbacks.py
index d730cf68..20d457d1 100644
--- a/src/ptbench/engine/callbacks.py
+++ b/src/ptbench/engine/callbacks.py
@@ -46,7 +46,7 @@ class LoggingCallback(Callback):
self.log("total_time", current_time)
self.log("eta", eta_seconds)
self.log("loss", numpy.average(self.training_loss))
- self.log("learning_rate", pl_module.lr)
+ self.log("learning_rate", pl_module.hparams["optimizer_params"]["lr"])
self.log("validation_loss", numpy.average(self.validation_loss))
queue_retries = 0
--
GitLab