bug fix

7 years ago · 7fbddaa64a
parent cb7c1245b3
commit 7fbddaa64a
1 changed files with 2 additions and 1 deletions
--- a/python/paddle/fluid/trainer.py
+++ b/python/paddle/fluid/trainer.py
@ -365,6 +365,7 @@ class Trainer(object):
            event_handler(BeginEpochEvent(epoch_id))
            for step_id, data in enumerate(reader()):
                if self.__stop:
+                    if self.checkpoint_cfg:
                        self._clean_checkpoint()
                    return