graphs.py 文件源码

def __init__(self, codes_shape, scipy_gmm):
        self.graph = tf.Graph()
        sess_config = tf.ConfigProto()
        sess_config.gpu_options.allow_growth = True
        self.sess = tf.Session(graph=self.graph, config=sess_config)

        print self.graph
        n,c = codes_shape

        with self.graph.as_default():
            self.pl = PlaceholderManager()
            self.pl.add_placeholder('codes', tf.float32, codes_shape)
            self.pl.add_placeholder('phase_train', tf.bool, [])

            with tf.variable_scope('GMM'):
                self.gmm = models.GMM(self.pl['codes'], scipy_gmm, self.pl['phase_train'])

        print var_collect.collect_all(self.graph)
        #tf.initialize_variables(
        #    var_list=var_collect.collect_all(self.graph)).run(session=self.sess)