added flush

williamFalcon · Jul 26, 2019 · 914bd9c · 914bd9c
1 parent 8fae0c3
commit 914bd9c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/test_tube/log.py b/test_tube/log.py
@@ -164,6 +164,7 @@ def get_meta_copy(self):
         return DDPExperiment(self)
 
     def on_exit(self):
+        self.__clean_dir()
         if self.rank == 0:
             self.close()