Log errors when doing eval.

PiperOrigin-RevId: 362205649

Log errors when doing eval.
PiperOrigin-RevId: 362205649
bee6a471 · Vighnesh Birodkar · TF Object Detection Team · 93da8e77 · bee6a471
Commit bee6a471 authored Mar 10, 2021 by Vighnesh Birodkar Committed by TF Object Detection Team Mar 10, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

research/object_detection/model_lib_v2.py research/object_detection/model_lib_v2.py +2 -1

No files found.
--- a/research/object_detection/model_lib_v2.py
+++ b/research/object_detection/model_lib_v2.py
@@ -896,7 +896,8 @@ def eager_eval_loop(
      (losses_dict, prediction_dict, groundtruth_dict,
       eval_features) = strategy.run(
           compute_eval_dict, args=(features, labels))
-    except:  # pylint:disable=bare-except
+    except Exception as exc:  # pylint:disable=broad-except
+      tf.logging.info('Encountered %s exception.', exc)
      tf.logging.info('A replica probably exhausted all examples. Skipping '
                      'pending examples on other replicas.')
      break