added flush

williamFalcon · Jul 26, 2019 · 52ff2bb · 52ff2bb
1 parent 17c4e78
commit 52ff2bb
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/test_tube/log.py b/test_tube/log.py
@@ -150,8 +150,7 @@ def __init__(
 
         # set the tensorboardx log path to the /tf folder in the exp folder
         logdir = self.get_tensorboardx_path(self.name, self.version)
-        if self.rank == 0:
-            super().__init__(log_dir=logdir, *args, **kwargs)
+        super().__init__(log_dir=logdir, *args, **kwargs)
 
         # register on exit fx so we always close the writer
         atexit.register(self.on_exit)