Add a new method SummaryWriter.flush() (#20607)

nataliakliushkina · facebook-github-bot · commit c0a2a3b22b54 · 2019-05-20T11:05:12.000-07:00
Summary: Pull Request resolved: #20607 Add a new method SummaryWriter.flush() that iterates through all of the FileWriters and flushes them Reviewed By: orionr Differential Revision: D15380124 fbshipit-source-id: 1975f3f61c5ae3754552bfdb23f2cd78f687d19f
diff --git a/torch/utils/tensorboard/writer.py b/torch/utils/tensorboard/writer.py
@@ -837,6 +837,16 @@ def add_custom_scalars(self, layout):
         """
         self._get_file_writer().add_summary(custom_scalars(layout))
 
+    def flush(self):
+        """Flushes the event file to disk.
+        Call this method to make sure that all pending events have been written to
+        disk.
+        """
+        if self.all_writers is None:
+            return
+        for writer in self.all_writers.values():
+            writer.flush()
+
     def close(self):
         if self.all_writers is None:
             return  # ignore double close
