added flush

williamFalcon · Jul 26, 2019 · 0350fb9 · 0350fb9
1 parent 8989b3c
commit 0350fb9
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/test_tube/log.py b/test_tube/log.py
@@ -512,6 +512,9 @@ def __hash__(self):
         return 'Exp: {}, v: {}'.format(self.name, self.version)
 
     def flush(self):
+        if self.rank == 0:
+            return
+
         if self.all_writers is None:
             return  # ignore double close