Skip to content
Snippets Groups Projects
Commit 41aa1b2a authored by Amir MOHAMMADI's avatar Amir MOHAMMADI
Browse files

Skip not found checkpoints

parent 422a72a6
No related branches found
No related tags found
3 merge requests!22Add a prediction script,!21Resolve "Adopt to the Estimators API",!20Create a default runconfig
Pipeline #
...@@ -91,7 +91,11 @@ def main(argv=None): ...@@ -91,7 +91,11 @@ def main(argv=None):
continue continue
for checkpoint_path in ckpt.all_model_checkpoint_paths: for checkpoint_path in ckpt.all_model_checkpoint_paths:
global_step = str(get_global_step(checkpoint_path)) try:
global_step = str(get_global_step(checkpoint_path))
except Exception:
print('Failed to find global_step for checkpoint_path {}, '
'skipping ...'.format(checkpoint_path))
if global_step in evaluated_steps: if global_step in evaluated_steps:
continue continue
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment