session.py 文件源码

def initialize_tf_variables(self):
        """
        Initialize tensorflow variables (either initializes them from scratch or restores from checkpoint).

        :return: updated TeLL session
        """
        session = self.tf_session
        checkpoint = self.workspace.get_checkpoint()
        #
        # Initialize or load variables
        #
        with Timer(name="Initializing variables"):
            session.run(tf.global_variables_initializer())
            session.run(tf.local_variables_initializer())

        if checkpoint is not None:
            # restore from checkpoint
            self.tf_saver.restore(session, checkpoint)
            # get step number from checkpoint
            step = session.run(self.__global_step_placeholder) + 1
            self.global_step = step
            # reopen summaries
            for _, summary in self.tf_summaries.items():
                summary.reopen()
                summary.add_session_log(tf.SessionLog(status=tf.SessionLog.START), global_step=step)
            print("Resuming from checkpoint '{}' at iteration {}".format(checkpoint, step))
        else:
            for _, summary in self.tf_summaries.items():
                summary.add_graph(session.graph)

        return self