[에러 해결] AttributeError: module 'tensorflow' has no attribute 'Summary'

github.com/taomanwai/tensorboardcolab/issues/9 를 참고했다.

 

 

 

def __init__(self, log_dir):
    """Initialize summary writer."""
    self.writer = tf.summary.FileWriter(log_dir)

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)

이 코드에서 에러가 난다. tensorflow 버전 차이로 인한 오류인 것 같다.

아래와 같이 수정하면 해결된다.

 

def __init__(self, log_dir):
    """Initialize summary writer."""
    self.writer = tf.summary.create_file_writer(log_dir)
        
def scalar_summary(self, tag, value, step):
    """Add scalar summary."""
    with self.writer.as_default():
        tf.summary.scalar(tag, value, step)
        self.writer.flush()