From f76a4398426cad6b131809e31e3f086c5b05f41e Mon Sep 17 00:00:00 2001
From: PabloNA97 <p.navarro.acero@gmail.com>
Date: Tue, 26 Aug 2025 10:50:21 +0200
Subject: [PATCH] log lr

---
 mlcolvar/cvs/unsupervised/vae.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/mlcolvar/cvs/unsupervised/vae.py b/mlcolvar/cvs/unsupervised/vae.py
index 7ff297b4..918488f5 100644
--- a/mlcolvar/cvs/unsupervised/vae.py
+++ b/mlcolvar/cvs/unsupervised/vae.py
@@ -229,6 +229,7 @@ def training_step(self, train_batch, batch_idx):
         # Log.
         name = "train" if self.training else "valid"
         self.log(f"{name}_loss", loss, on_epoch=True)
+        self.log("lr", self.optimizers().param_groups[0]['lr'], on_epoch=True)
 
         return loss