def end(self, session):
    last_step = session.run(contrib_variables.get_global_step())
    self._save(last_step, session)
def end(self, session):
    last_step = session.run(contrib_variables.get_global_step())
    self._save(last_step, session)