Skip to content

Commit 53409a2

Browse files
committed
code optimized
1 parent f28f41d commit 53409a2

File tree

1 file changed

+5
-9
lines changed

1 file changed

+5
-9
lines changed

python/paddle/fluid/trainer.py

Lines changed: 5 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -141,14 +141,10 @@ def __init__(self,
141141
self.chief = True
142142
self.checkpoint = checkpoint_config
143143
if self.checkpoint:
144-
if not isinstance(self.checkpoint, CheckpointConfig):
145-
raise TypeError(
146-
"The checkpoint_config shoule be an instance of CheckpointConfig"
147-
)
148-
else:
149-
serial = io.get_latest_checkpoint_serial(
150-
self.checkpoint.checkpoint_dir)
151-
self.checkpoint.load_serial = serial if serial >= 0 else None
144+
assert isinstance(self.checkpoint, CheckpointConfig)
145+
serial = io.get_latest_checkpoint_serial(
146+
self.checkpoint.checkpoint_dir)
147+
self.checkpoint.load_serial = serial if serial >= 0 else None
152148

153149
self.scope = core.Scope()
154150

@@ -385,8 +381,8 @@ def _train_by_any_executor(self, event_handler, exe, num_epochs, reader):
385381
else:
386382
metrics = exe.run(feed=data, fetch_list=[])
387383

388-
event_handler(EndStepEvent(epoch_id, step_id, metrics))
389384
self._save_checkpoint(epoch_id, step_id)
385+
event_handler(EndStepEvent(epoch_id, step_id, metrics))
390386
event_handler(EndEpochEvent(epoch_id))
391387
self._clean_checkpoint()
392388

0 commit comments

Comments
 (0)