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/learn | |
parent | a738649f8092dda124b6a4f3ccc31bf4159651ea (diff) |
Support distribution strategies in `Estimator.evaluate`.
PiperOrigin-RevId: 206864512
Diffstat (limited to 'tensorflow/contrib/learn')
-rw-r--r-- | tensorflow/contrib/learn/python/learn/estimators/run_config.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/contrib/learn/python/learn/estimators/run_config.py b/tensorflow/contrib/learn/python/learn/estimators/run_config.py index 7cb87619d9..c36879e048 100644 --- a/tensorflow/contrib/learn/python/learn/estimators/run_config.py +++ b/tensorflow/contrib/learn/python/learn/estimators/run_config.py @@ -302,6 +302,7 @@ class RunConfig(ClusterConfig, core_run_config.RunConfig): # so instead of breaking compatibility with that assumption, we # just manually initialize this field: self._train_distribute = None + self._eval_distribute = None self._device_fn = None gpu_options = config_pb2.GPUOptions( |