diff --git a/bob/learn/tensorflow/script/eval_generic.py b/bob/learn/tensorflow/script/eval_generic.py index f29f756707c3c643711fbb6de9062dd3adb60aba..de7701663ac8b92d73cbe805b17005d2d13b1caa 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