tf.train.ClusterSpec

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

tf.train.ClusterSpec(
    cluster
)

Объект 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"]})

Методы

as_cluster_def

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

as_cluster_def()

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

as_dict

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

as_dict()

Возвращает словарь от имён работ до их задач.

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

job_tasks

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

job_tasks(
    job_name
)

Возвращает отображение идентификатора задачи на адрес в заданной работе.

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

num_tasks

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

num_tasks(
    job_name
)

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

task_address

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

task_address(
    job_name, task_index
)

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

task_indices

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

task_indices(
    job_name
)

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

__bool__

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

__bool__()

__eq__

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

__eq__(
    other
)

Возвращает self==value.

__ne__

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

__ne__(
    other
)

Возвращает self!=value.

__nonzero__

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

__nonzero__()