Skip to content

Commit e901de6

Browse files
committed
update var name
1 parent 2412dee commit e901de6

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

python/paddle/fluid/io.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323
__all__ = [
2424
'save_vars', 'save_params', 'save_persistables', 'load_vars', 'load_params',
2525
'load_persistables', 'save_inference_model', 'load_inference_model',
26-
'get_inference_program', 'save_checkpoint', 'restore_checkpoint'
26+
'get_inference_program', 'save_checkpoint', 'load_checkpoint'
2727
]
2828

2929

@@ -466,7 +466,7 @@ def save_checkpoint(executor,
466466
Save Variables to Checkpoint Directory
467467
468468
:param dirname
469-
:param keep_max
469+
:param max_num_checkpoints
470470
:param save_secs
471471
:param main_program
472472
"""
@@ -495,7 +495,7 @@ def save_checkpoint(executor,
495495
_lru_delete(dirname, max_num_checkpoints)
496496

497497

498-
def restore_checkpoint(executor, dirname=None, main_program=None):
498+
def load_checkpoint(executor, dirname=None, main_program=None):
499499
"""
500500
Load Variables from Checkpint Dir
501501
@@ -544,9 +544,9 @@ def _interval_secs_exceed(dirname, save_interval_secs):
544544
return True
545545

546546

547-
def _lru_delete(dirname, keep_max=3):
547+
def _lru_delete(dirname, max_num_checkpoints=3):
548548
"""
549-
retain checkpoint nums with keep_max
549+
retain checkpoint nums with max_num_checkpoints
550550
"""
551551
dirs = os.listdir(dirname)
552552
serials = []
@@ -556,11 +556,11 @@ def _lru_delete(dirname, keep_max=3):
556556
except ValueError:
557557
continue
558558

559-
if len(serials) <= keep_max:
559+
if len(serials) <= max_num_checkpoints:
560560
return
561561

562562
serials.sort(reverse=True)
563-
serials = serials[keep_max:]
563+
serials = serials[max_num_checkpoints:]
564564
for serial in serials:
565565
cur_dir = os.path.join(dirname, str(serial))
566566
shutil.rmtree(cur_dir)

0 commit comments

Comments
 (0)