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

Added chief to the default device_filters for /job:ps

Browse Source 
This behaviour matches the description in the
_get_default_session_config_distributed docstring, and restores the
symmetry of the default device_filters.
This commit is contained in:
Sergei Lebedev 2018-09-27 16:52:51 +02:00
parent 19d8963bc0
commit 4c0c4bb3fe
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:
device_filters = ['/job:ps', '/job:worker/task:%d' % self._task_id]
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:
# If the task_type is `EVALUATOR` or something other than the ones in
# 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)
self._assert_equal_session_config(run_config.session_config,
['/job:ps', '/job:worker', '/job:master'])
self._assert_equal_session_config(
run_config.session_config,
['/job:ps', '/job:worker', '/job:chief', '/job:master'])
def test_evaluator_session_config(self):
tf_config = {