def METHOD_NAME(self, trainer, inner_iter=False):
    """Get the current training iteration step."""
    if self.by_epoch and inner_iter:
        current_iter = trainer.inner_iter + 1
    else:
        current_iter = trainer.iter + 1
    return current_iter