def on_train_epoch_start(self, trainer: pl.Trainer, pl_module: pl.LightningModule):
    pl_module.log(self.target_steps_attr, float(lightning_getattr(pl_module, self.target_steps_attr)), on_step=False, on_epoch=True)
def on_train_epoch_start(self, trainer: pl.Trainer, pl_module: pl.LightningModule):
    pl_module.log(self.target_steps_attr, float(lightning_getattr(pl_module, self.target_steps_attr)), on_step=False, on_epoch=True)