Skip to content

Commit 08e5f0a

Browse files
committed
rename need_load_checkpoint to get_latest_checkpoint_serial
1 parent c06f43b commit 08e5f0a

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

python/paddle/fluid/io.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@
2525
'load_persistables', 'save_inference_model', 'load_inference_model',
2626
'get_inference_program', 'save_checkpoint', 'load_checkpoint',
2727
'clean_checkpoint', 'load_persist_vars_without_grad',
28-
'save_persist_vars_without_grad'
28+
'save_persist_vars_without_grad', 'get_latest_checkpoint_serial'
2929
]
3030

3131

@@ -503,7 +503,7 @@ def save_checkpoint(executor,
503503
_lru_delete(checkpoint_dir, max_num_checkpoints)
504504

505505

506-
def need_load_checkpoint(checkpoint_dir):
506+
def get_latest_checkpoint_serial(checkpoint_dir):
507507
"""
508508
If the directory have checkpoint files, it will return lastest checkpoint directory serial number
509509

python/paddle/fluid/trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -146,7 +146,7 @@ def __init__(self,
146146
"The checkpoint_config shoule be an instance of CheckpointConfig"
147147
)
148148
else:
149-
self.checkpoint.load_serial = io.need_load_checkpoint(
149+
self.checkpoint.load_serial = io.get_latest_checkpoint_serial(
150150
self.checkpoint.checkpoint_dir)
151151

152152
self.scope = core.Scope()

0 commit comments

Comments
 (0)