def get_step(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)
Get the total training step/epoch.
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/runner/hooks/logger/pavi.py