Merge pull request #1 from rra94/rra94-patch-1

Error with latest tensorflow
rra94 · May 3, 2020 · 05e4afc · 05e4afc
2 parents 94dd002 + d54f584
commit 05e4afc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/logger.py b/logger.py
@@ -6,9 +6,9 @@ class Logger(object):
 
     def __init__(self, log_dir):
         """Initialize summary writer."""
-        self.writer = tf.summary.FileWriter(log_dir)
+        self.writer = tf.summary.create_file_writer(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)
+        self.writer.add_summary(summary, step)