changed on_exit

williamFalcon · Sep 10, 2019 · 72eb9ba · 72eb9ba
1 parent 14fa365
commit 72eb9ba
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/test_tube/log.py b/test_tube/log.py
@@ -164,10 +164,12 @@ def get_meta_copy(self):
         return DDPExperiment(self)
 
     def on_exit(self):
-        self.__clean_dir()
         if self.rank == 0:
             self.close()
 
+        self.__clean_dir()
+
+
     def __clean_dir(self):
         files = os.listdir(self.save_dir)
 
@@ -176,7 +178,6 @@ def __clean_dir(self):
 
         for f in files:
             if str(self.process) in f:
-                self.close()
                 os.remove(os.path.join(self.save_dir, f))
 
     def argparse(self, argparser):