'Saves the session, restores variables from checkpoint.
Args:
session: A tensorflow session.'

def initialize_and_restore(self, session):

    self.session = session
    self.restore_vars_from_checkpoint(self.checkpoint_dir)
