File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change 25
25
'load_persistables' , 'save_inference_model' , 'load_inference_model' ,
26
26
'get_inference_program' , 'save_checkpoint' , 'load_checkpoint' ,
27
27
'clean_checkpoint' , 'load_persist_vars_without_grad' ,
28
- 'save_persist_vars_without_grad'
28
+ 'save_persist_vars_without_grad' , 'get_latest_checkpoint_serial'
29
29
]
30
30
31
31
@@ -503,7 +503,7 @@ def save_checkpoint(executor,
503
503
_lru_delete (checkpoint_dir , max_num_checkpoints )
504
504
505
505
506
- def need_load_checkpoint (checkpoint_dir ):
506
+ def get_latest_checkpoint_serial (checkpoint_dir ):
507
507
"""
508
508
If the directory have checkpoint files, it will return lastest checkpoint directory serial number
509
509
Original file line number Diff line number Diff line change @@ -146,7 +146,7 @@ def __init__(self,
146
146
"The checkpoint_config shoule be an instance of CheckpointConfig"
147
147
)
148
148
else :
149
- self .checkpoint .load_serial = io .need_load_checkpoint (
149
+ self .checkpoint .load_serial = io .get_latest_checkpoint_serial (
150
150
self .checkpoint .checkpoint_dir )
151
151
152
152
self .scope = core .Scope ()
You can’t perform that action at this time.
0 commit comments