tf.train.ClusterSpec

Представляет кластер как набор «задач», объединённых в «задания».

Просмотреть псевдонимы

Псевдонимы совместимости для миграции

Подробнее см. в руководстве по миграции.

tf.compat.v1.train.ClusterSpec

tf.train.ClusterSpec(
    cluster
)

Используется в блокнотах

Используется в руководствах
Пользовательский цикл обучения с Keras и MultiWorkerMirroredStrategy Обучение с сервером параметров и ParameterServerStrategy

tf.train.ClusterSpec представляет набор процессов, участвующих в распределённых вычислениях TensorFlow. Каждый tf.distribute.Server создаётся в определённом кластере.

Чтобы создать кластер с двумя заданиями и пятью задачами, укажите соответствие имён заданий спискам сетевых адресов (обычно это пары «имя хоста — порт»).

cluster = tf.train.ClusterSpec({"worker": ["worker0.example.com:2222",
                                           "worker1.example.com:2222",
                                           "worker2.example.com:2222"],
                                "ps": ["ps0.example.com:2222",
                                       "ps1.example.com:2222"]})

Каждое задание также можно указать как разреженное соответствие индексов задач сетевым адресам. Это позволяет настроить сервер, не зная идентификаторы (например) всех остальных задач рабочих узлов:

cluster = tf.train.ClusterSpec({"worker": {1: "worker1.example.com:2222"},
                                "ps": ["ps0.example.com:2222",
                                       "ps1.example.com:2222"]})

Аргументы
`cluster`	Словарь, сопоставляющий одно или несколько имён заданий с (i) списком сетевых адресов или (ii) словарём, сопоставляющим целочисленные индексы задач с сетевыми адресами; либо буфер протокола `tf.train.ClusterDef`.

Исключения
`TypeError`	Если `cluster` не является словарём, сопоставляющим строки со списками строк, и не является протобуфером `tf.train.ClusterDef`.

Атрибуты
`jobs`	Возвращает список имён заданий в этом кластере.

Методы

`as_cluster_def`

Посмотреть исходный код

as_cluster_def()

Возвращает буфер протокола tf.train.ClusterDef, созданный на основе этого кластера.

`as_dict`

Посмотреть исходный код

as_dict()

Возвращает словарь, сопоставляющий имена заданий их задачам.

Для каждого задания, если пространство индексов задач плотное, соответствующим значением будет список сетевых адресов; в противном случае это будет словарь, сопоставляющий (разреженные) индексы задач соответствующим адресам.

Возвращает
Словарь, сопоставляющий имена заданий спискам или словарям с описанием задач в этих заданиях.

`job_tasks`

Посмотреть исходный код

job_tasks(
    job_name
)

Возвращает соответствие идентификаторов задач адресам в указанном задании.

Примечание: Для обеспечения обратной совместимости этот метод возвращает список. Если указанное задание определено с разреженным набором индексов задач, длина этого списка может не соответствовать количеству задач, определённых в задании. Чтобы узнать количество задач, определённых в конкретном задании, используйте метод tf.train.ClusterSpec.num_tasks.

Аргументы
`job_name`	Строковое имя задания в этом кластере.

Возвращает
Список адресов задач, где индекс элемента списка соответствует индексу каждой задачи. Список может содержать `None`, если задание определено с разреженным набором индексов задач.

Исключения
`ValueError`	Если `job_name` не обозначает задание в этом кластере.

`num_tasks`

Посмотреть исходный код

num_tasks(
    job_name
)

Возвращает количество задач, определённых в указанном задании.

Аргументы
`job_name`	Строковое имя задания в этом кластере.

Возвращает
Количество задач, определённых в указанном задании.

Исключения
`ValueError`	Если `job_name` не обозначает задание в этом кластере.

`task_address`

Посмотреть исходный код

task_address(
    job_name, task_index
)

Возвращает адрес указанной задачи в указанном задании.

Аргументы
`job_name`	Строковое имя задания в этом кластере.
`task_index`	Неотрицательное целое число.

Возвращает
Адрес указанной задачи в указанном задании.

Исключения
`ValueError`	Если `job_name` не обозначает задание в этом кластере или в этом задании не определена задача с индексом `task_index`.

`task_indices`

Посмотреть исходный код

task_indices(
    job_name
)

Возвращает список допустимых индексов задач в указанном задании.

Аргументы
`job_name`	Строковое имя задания в этом кластере.

Возвращает
Список допустимых индексов задач в указанном задании.

Исключения
`ValueError`	Если `job_name` не обозначает задание в этом кластере или в этом задании не определена задача с индексом `task_index`.

`bool`

Посмотреть исходный код

__bool__()

`eq`

Посмотреть исходный код

__eq__(
    other
)

Возвращает результат сравнения self==value.

`ne`

Посмотреть исходный код

__ne__(
    other
)

Возвращает результат сравнения self!=value.

`nonzero`

Посмотреть исходный код

__nonzero__()

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/train/ClusterSpec