model_session.py 文件源码

def create(cls, **kwargs):
        """
        Create a new model session.

        :param kwargs: optional graph parameters
        :type kwargs: dict
        :return: new model session
        :rtype: ModelSession
        """
        session = tf.Session()
        with session.graph.as_default():
            cls.create_graph(**kwargs)
        session.run(tf.initialize_all_variables())
        return cls(session, tf.train.Saver())