aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/training/python/training/evaluation.py
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/contrib/training/python/training/evaluation.py')
-rw-r--r--tensorflow/contrib/training/python/training/evaluation.py3
1 files changed, 1 insertions, 2 deletions
diff --git a/tensorflow/contrib/training/python/training/evaluation.py b/tensorflow/contrib/training/python/training/evaluation.py
index c14f4ee440..212f2c7a7b 100644
--- a/tensorflow/contrib/training/python/training/evaluation.py
+++ b/tensorflow/contrib/training/python/training/evaluation.py
@@ -410,8 +410,7 @@ def evaluate_repeatedly(checkpoint_dir,
'%Y-%m-%d-%H:%M:%S', time.gmtime()))
num_evaluations += 1
- reached_max = num_evaluations >= max_number_of_evaluations
- if max_number_of_evaluations and reached_max:
+ if max_number_of_evaluations is not None and num_evaluations >= max_number_of_evaluations:
return final_ops_hook.final_ops_values
logging.info('Timed-out waiting for a checkpoint.')