def step_begin(self, step):
    super(CheckpointSaver, self).step_begin(step)
    self._last_begin_step = step
def step_begin(self, step):
    super(CheckpointSaver, self).step_begin(step)
    self._last_begin_step = step