diff --git a/src/graphnet/models/standard_model.py b/src/graphnet/models/standard_model.py index 663664996..53c32deaf 100644 --- 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(