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 #
This commit is part of merge request !22. Comments created here will be created in the context of that merge request.
......@@ -91,7 +91,11 @@ def main(argv=None):
continue
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:
continue
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment