diff options
author | Priya Gupta <priyag@google.com> | 2018-07-31 18:41:28 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-31 18:45:54 -0700 |
commit | ffc9be66bc1503ec12f10f43ff622d9193062e97 (patch) | |
tree | d74f8057b6dd391ceb533db7a8decd8b0d88e86e /tensorflow/contrib/estimator | |
parent | a738649f8092dda124b6a4f3ccc31bf4159651ea (diff) |
Support distribution strategies in `Estimator.evaluate`.
PiperOrigin-RevId: 206864512
Diffstat (limited to 'tensorflow/contrib/estimator')
-rw-r--r-- | tensorflow/contrib/estimator/python/estimator/saved_model_estimator.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/estimator/python/estimator/saved_model_estimator.py b/tensorflow/contrib/estimator/python/estimator/saved_model_estimator.py index b0082f7e55..ce98e9987e 100644 --- a/tensorflow/contrib/estimator/python/estimator/saved_model_estimator.py +++ b/tensorflow/contrib/estimator/python/estimator/saved_model_estimator.py @@ -148,7 +148,7 @@ class SavedModelEstimator(estimator_lib.Estimator): super(SavedModelEstimator, self).__init__( model_fn=self._model_fn_from_saved_model, model_dir=model_dir, warm_start_from=warm_start_settings) - if self._distribution is not None: + if self._train_distribution or self._eval_distribution: raise NotImplementedError( 'SavedModelEstimator currently does not support ' 'DistributionStrategy.') |