logger.py

import tensorflow as tf


class Logger(object):
    def __init__(self, log_dir):
        """Create a summary writer logging to log_dir."""
        self.writer = tf.summary.create_file_writer(log_dir)

    def scalar_summary(self, tag, value, step):
        """Log a scalar variable."""
        with self.writer.as_default():
            tf.summary.scalar(tag, value, step=step)
            self.writer.flush()
        # summary = tf.Summary(value=[tf.Summary.Value(tag=tag, simple_value=value)])
        # self.writer.add_summary(summary, step)

    # def scalar_summary(self, tag, value, step):
    #     """Add scalar summary."""
    #     summary = tf.summary(value=[tf.summary.Value(tag=tag, simple_value=value)])
    #     self.writer.add_summary(summary, step)