diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-03-28 21:52:30 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-03-28 21:54:59 -0700 |
commit | 163bf8d0620a08d186c1315b0789e898f09759f8 (patch) | |
tree | 8aa97197db02f33002bcdae2f94aea5d8b3cd6c5 /tensorflow/contrib/learn | |
parent | aeaec465f2f08e32c524e23fb7b0ac016f3dc6a9 (diff) |
DistributionStrategy-enable Estimator.
PiperOrigin-RevId: 190882152
Diffstat (limited to 'tensorflow/contrib/learn')
-rw-r--r-- | tensorflow/contrib/learn/python/learn/estimators/run_config.py | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/tensorflow/contrib/learn/python/learn/estimators/run_config.py b/tensorflow/contrib/learn/python/learn/estimators/run_config.py index 1d161093de..f3500bf56f 100644 --- a/tensorflow/contrib/learn/python/learn/estimators/run_config.py +++ b/tensorflow/contrib/learn/python/learn/estimators/run_config.py @@ -290,8 +290,15 @@ class RunConfig(ClusterConfig, core_run_config.RunConfig): Note - using this argument, it is easy to provide settings which break otherwise perfectly good models. Use with care. """ - super(RunConfig, self).__init__( - master=master, evaluation_master=evaluation_master) + # Neither parent class calls super().__init__(), so here we have to + # manually call their __init__() methods. + ClusterConfig.__init__( + self, master=master, evaluation_master=evaluation_master) + # For too long this code didn't call: + # core_run_config.RunConfig.__init__(self) + # so instead of breaking compatibility with that assumption, we + # just manually initialize this field: + self._distribute = None gpu_options = config_pb2.GPUOptions( per_process_gpu_memory_fraction=gpu_memory_fraction) |