Skip to content

Commit 7fbddaa

Browse files
committed
bug fix
1 parent cb7c124 commit 7fbddaa

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

python/paddle/fluid/trainer.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -365,7 +365,8 @@ def _train_by_any_executor(self, event_handler, exe, num_epochs, reader):
365365
event_handler(BeginEpochEvent(epoch_id))
366366
for step_id, data in enumerate(reader()):
367367
if self.__stop:
368-
self._clean_checkpoint()
368+
if self.checkpoint_cfg:
369+
self._clean_checkpoint()
369370
return
370371

371372
if self.checkpoint_cfg and self.checkpoint_cfg.load_serial \

0 commit comments

Comments
 (0)