Merge pull request #21 from RasmusOrsoe/main

update from main
graphnet-team · Feb 8, 2024 · 7c4fdfb · 7c4fdfb
2 parents 36a2384 + 5c93a7a
commit 7c4fdfb
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/graphnet/models/standard_model.py b/src/graphnet/models/standard_model.py
@@ -273,6 +273,9 @@ def training_step(
             on_step=False,
             sync_dist=True,
         )
+
+        current_lr = self.trainer.optimizers[0].param_groups[0]["lr"]
+        self.log("lr", current_lr, prog_bar=True, on_step=True)
         return loss
 
     def validation_step(