dnn.py 文件源码

def buildSummaryGraph(self):
        """
            TODO: Have to fix summary update in case of discriminative pre-training
        """
        # TODO: Have to fix summary update in case of discriminative pre - training
        self.summaryWriter = tf.train.SummaryWriter('logdir5', tf.get_default_graph())
        tf.scalar_summary(self.loss.op.name, self.loss)
        tf.scalar_summary(self.learning_rate.op.name, self.learning_rate)
        self.summary = tf.merge_all_summaries()