Exit eval if training steps are complete.

PiperOrigin-RevId: 428629058

Exit eval if training steps are complete.
PiperOrigin-RevId: 428629058
8c3a1ef3 · Vighnesh Birodkar · TF Object Detection Team · 812ab923 · 8c3a1ef3
Commit 8c3a1ef3 authored Feb 14, 2022 by Vighnesh Birodkar Committed by TF Object Detection Team Feb 14, 2022
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

research/object_detection/model_lib_v2.py research/object_detection/model_lib_v2.py +4 -0

No files found.
--- a/research/object_detection/model_lib_v2.py
+++ b/research/object_detection/model_lib_v2.py
@@ -1164,3 +1164,7 @@ def eval_continuously(
          postprocess_on_cpu=postprocess_on_cpu,
          global_step=global_step,
          )
+
+    if global_step.numpy() == configs['train_config'].num_steps:
+      tf.logging.info('Exiting evaluation at step %d', global_step.numpy())
+      return