experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #22566 from superbobry:default-session-config-chief

Browse Source 
PiperOrigin-RevId: 217439914
This commit is contained in:
TensorFlower Gardener 2018-10-16 21:55:14 -07:00
commit ea718e872d
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/python/estimator/run_config.py
View File

@ -566,7 +566,7 @@ class RunConfig(object):
elif self._task_type == TaskType.WORKER: elif self._task_type == TaskType.WORKER:
device_filters = ['/job:ps', '/job:worker/task:%d' % self._task_id] device_filters = ['/job:ps', '/job:worker/task:%d' % self._task_id]
elif self._task_type == TaskType.PS: elif self._task_type == TaskType.PS:
device_filters = ['/job:ps', '/job:worker', '/job:master'] device_filters = ['/job:ps', '/job:worker', '/job:chief', '/job:master']
else: else:
# If the task_type is `EVALUATOR` or something other than the ones in # If the task_type is `EVALUATOR` or something other than the ones in
# TaskType then don't set any device filters. # TaskType then don't set any device filters.

5
tensorflow/python/estimator/run_config_test.py
View File

@ -1196,8 +1196,9 @@ class RunConfigSessionConfigTest(test.TestCase):
} }
} }
run_config = _create_run_config_with_cluster_spec(tf_config) run_config = _create_run_config_with_cluster_spec(tf_config)
self._assert_equal_session_config(run_config.session_config, self._assert_equal_session_config(
['/job:ps', '/job:worker', '/job:master']) run_config.session_config,
['/job:ps', '/job:worker', '/job:chief', '/job:master'])
def test_evaluator_session_config(self): def test_evaluator_session_config(self):
tf_config = { tf_config = {