TensorFlow2.0 module ‘tensorflow‘ has no attribute ‘Summary‘

最新推荐文章于 2024-03-04 23:43:06 发布

PJlei

最新推荐文章于 2024-03-04 23:43:06 发布

阅读量9.4k

点赞数 18

分类专栏： python基础

本文链接：https://blog.youkuaiyun.com/zl1107604962/article/details/109441038

版权

python基础专栏收录该内容

4 篇文章

订阅专栏

使用TensorBoard过程中报错 module ‘tensorflow’ has no attribute ‘Summary’
修改成如下代码就可以了

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

    def scalar_summary(self, tag, value, step):
        """Log a scalar variable."""
        #summary = tf.Summary(value=[tf.Summary.Value(tag=tag, simple_value=value)])
        #self.writer.add_summary(summary, step)
        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 list_of_scalars_summary(self, tag_value_pairs, step):
        """Log scalar variables."""
        # summary = tf.Summary(value=[tf.Summary.Value(tag=tag, simple_value=value) for tag, value in tag_value_pairs])
        #self.writer.add_summary(summary, step)
        with self.writer.as_default():
            for tag, value in tag_value_pairs:
                tf.summary.scalar(tag, value, step=step)
            self.writer.flush()
        # summary = tf.Summary(value=[tf.Summary.Value(tag=tag, simple_value=value) for tag, value in tag_value_pairs])
        # self.writer.add_summary(summary, step)