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

Skip not found checkpoints

parent 422a72a6
Branches
Tags
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):
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