def METHOD_NAME(self, runner):
    """Get the total training step/epoch."""
    if self.get_mode(runner) == 'val' and self.by_epoch:
        return self.get_epoch(runner)
    else:
        return self.get_iter(runner)