提交 e44c278e 编写于 作者: T tangwei12

bug fix about clean

上级 94eaf94c
...@@ -425,7 +425,7 @@ class Trainer(object): ...@@ -425,7 +425,7 @@ class Trainer(object):
return self._get_parallel_executor() return self._get_parallel_executor()
def _clean_checkpoint(self): def _clean_checkpoint(self):
if not self.checkpoint: if not self.checkpoint and not self.chief:
return return
io.clean_checkpoint(checkpoint_dir=self.checkpoint.checkpoint_dir) io.clean_checkpoint(checkpoint_dir=self.checkpoint.checkpoint_dir)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册