From 41aa1b2aa7397f495a8af3e14b7467f9c071b730 Mon Sep 17 00:00:00 2001
From: Amir MOHAMMADI <amir.mohammadi@idiap.ch>
Date: Fri, 13 Oct 2017 14:23:50 +0200
Subject: [PATCH] Skip not found checkpoints
---
bob/learn/tensorflow/script/eval_generic.py | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/bob/learn/tensorflow/script/eval_generic.py b/bob/learn/tensorflow/script/eval_generic.py
index f29f7567..de770166 100644
--- a/bob/learn/tensorflow/script/eval_generic.py
+++ b/bob/learn/tensorflow/script/eval_generic.py
@@ -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
--
GitLab