cancle only chief delete files

7 years ago · bca4da4225
parent e44c278e60
commit bca4da4225
2 changed files with 3 additions and 2 deletions
--- a/python/paddle/fluid/io.py
+++ b/python/paddle/fluid/io.py
@ -499,7 +499,8 @@ def save_checkpoint(executor,

    if is_chief:
        save_persist_vars_without_grad(executor, cur_dir, main_program)
-        _lru_delete(checkpoint_dir, max_num_checkpoints)
+
+    _lru_delete(checkpoint_dir, max_num_checkpoints)


 def need_load_checkpoint(checkpoint_dir):
--- a/python/paddle/fluid/trainer.py
+++ b/python/paddle/fluid/trainer.py
@ -425,7 +425,7 @@ class Trainer(object):
        return self._get_parallel_executor()

    def _clean_checkpoint(self):
-        if not self.checkpoint and not self.chief:
+        if not self.checkpoint:
            return
        io.clean_checkpoint(checkpoint_dir=self.checkpoint.checkpoint_dir)