diff options
| author |  Sourabh Bajaj <sourabhbajaj@google.com> | 2018-09-19 17:48:07 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-19 17:52:30 -0700 |
| commit | 415455b0ef2d65504ab8c9084a6daa2899521212 (patch) | |
| tree | 982e22160039f14d7c1b27ca2280501e875c89cd /tensorflow/contrib/tpu | |
| parent | 37632e5f0f8b0776854746152a3a7b361414f2cc (diff) | |
Fix the error message thrown when running eval on pod
PiperOrigin-RevId: 213730668
Diffstat (limited to 'tensorflow/contrib/tpu')
| -rw-r--r-- | tensorflow/contrib/tpu/python/tpu/tpu_context.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/tpu/python/tpu/tpu_context.py b/tensorflow/contrib/tpu/python/tpu/tpu_context.py index 19359cb612..ac76712aeb 100644 --- a/tensorflow/contrib/tpu/python/tpu/tpu_context.py +++ b/tensorflow/contrib/tpu/python/tpu/tpu_context.py @@ -599,8 +599,8 @@ class _InternalTPUContext(object): .format(self._eval_batch_size, num_replicas)) if num_hosts > 1 and not self.is_input_broadcast_with_iterators(): raise ValueError( - 'TPUEstimator.evaluate should be running on single TPU worker. ' - 'got {}.'.format(num_hosts)) + 'TPUEstimator.evaluate should be running on single TPU' + ' instead of a Pod.') else: assert mode == model_fn_lib.ModeKeys.PREDICT if self._predict_batch_size is None: |