def on_train_epoch_end(self, trainer: pl.Trainer, pl_module: pl.LightningModule):
    prev_target_steps = lightning_getattr(pl_module, self.target_steps_attr)
    target_steps = (prev_target_steps + self.increase)
    if (self.initial_target_steps is not None):
        target_steps = min(target_steps, self.initial_target_steps)
    if (target_steps != prev_target_steps):
        lightning_setattr(pl_module, self.target_steps_attr, target_steps)
        trainer.reset_train_dataloader(pl_module)
def on_train_epoch_end(self, trainer: pl.Trainer, pl_module: pl.LightningModule):
    prev_target_steps = lightning_getattr(pl_module, self.target_steps_attr)
    target_steps = (prev_target_steps + self.increase)
    if (self.initial_target_steps is not None):
        target_steps = min(target_steps, self.initial_target_steps)
    if (target_steps != prev_target_steps):
        lightning_setattr(pl_module, self.target_steps_attr, target_steps)
        trainer.reset_train_dataloader(pl_module)