diff options
| author |  Priya Gupta <priyag@google.com> | 2018-09-04 20:48:45 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-04 20:53:25 -0700 |
| commit | 734214903cfa8df6d55d25a04748b0989428f2ee (patch) | |
| tree | dd922b43ebf095907102d6bb399a81c77fb175eb /tensorflow/contrib/distribute/python/mirrored_strategy.py | |
| parent | e9332539bea372f6dbe6ef185f9d8b1f3b6e1fe2 (diff) | |
Set session_config.isolate_session_state to True for all strategies except Parameter server strategy where variables are shared across sessions.
PiperOrigin-RevId: 211573447
Diffstat (limited to 'tensorflow/contrib/distribute/python/mirrored_strategy.py')
| -rw-r--r-- | tensorflow/contrib/distribute/python/mirrored_strategy.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/contrib/distribute/python/mirrored_strategy.py b/tensorflow/contrib/distribute/python/mirrored_strategy.py index d1235b7afb..0c6805d682 100644 --- a/tensorflow/contrib/distribute/python/mirrored_strategy.py +++ b/tensorflow/contrib/distribute/python/mirrored_strategy.py @@ -572,6 +572,10 @@ class MirroredStrategy(distribute_lib.DistributionStrategy): task_type=None, task_id=None): del task_type, task_id + + if session_config: + session_config.isolate_session_state = True + if cluster_spec: self._initialize_multi_worker(self._num_gpus, cluster_spec) |