tf.distribute.Server
Сервер TensorFlow, работающий внутри процесса, предназначенный для распределённого обучения.
tf.distribute.Server(
server_or_cluster_def,
job_name=None,
task_index=None,
protocol=None,
config=None,
start=True
)
Используется в ноутбуках
| Используется в руководстве | Используется в учебных материалах |
|---|---|
Экземпляр tf.distribute.Server объединяет набор устройств и целевой объект tf.compat.v1.Session, которые могут участвовать в распределённом обучении. Сервер принадлежит кластеру (задаваемому с помощью tf.train.ClusterSpec) и соответствует определённой задаче в именованном задании. Сервер может взаимодействовать с любым другим сервером в том же кластере.
| Аргументы | |
|---|---|
server_or_cluster_def | Протокольный буфер tf.train.ServerDef или tf.train.ClusterDef либо объект tf.train.ClusterSpec, описывающий создаваемый сервер и/или кластер, членом которого он является. |
job_name | (Необязательно.) Указывает имя задания, членом которого является сервер. По умолчанию используется значение из server_or_cluster_def, если оно указано. |
task_index | (Необязательно.) Указывает индекс задачи сервера в его задании. По умолчанию используется значение из server_or_cluster_def, если оно указано. В противном случае по умолчанию используется 0, если в задании сервера только одна задача. |
protocol | (Необязательно.) Указывает протокол, который будет использоваться сервером. Допустимые значения включают "grpc", "grpc+verbs". По умолчанию используется значение из server_or_cluster_def, если оно указано. В противном случае по умолчанию используется "grpc". |
config | (Необязательно.) Объект tf.compat.v1.ConfigProto, задающий параметры конфигурации по умолчанию для всех сеансов, выполняемых на этом сервере. |
start | (Необязательно.) Логическое значение, указывающее, следует ли запускать сервер после его создания. По умолчанию — True. |
| Вызывает исключения | |
|---|---|
tf.errors.OpError | Или один из его подклассов, если при создании сервера TensorFlow произошла ошибка. |
| Атрибуты | |
|---|---|
server_def | Возвращает tf.train.ServerDef для этого сервера. |
target | Возвращает целевой объект, к которому может подключиться tf.compat.v1.Session. Чтобы создать server = tf.distribute.Server(...) with tf.compat.v1.Session(server.target): # ... |
Методы
create_local_server
@staticmethod
create_local_server(
config=None, start=True
)
Создаёт новый однопроцессный кластер, работающий на локальном хосте.
Этот метод служит упрощённой обёрткой для создания tf.distribute.Server с tf.train.ServerDef, который задаёт однопроцессный кластер с одной задачей в задании с именем "local".
| Аргументы | |
|---|---|
config | (Необязательно.) Объект tf.compat.v1.ConfigProto, задающий параметры конфигурации по умолчанию для всех сеансов, выполняемых на этом сервере. |
start | (Необязательно.) Логическое значение, указывающее, следует ли запускать сервер после его создания. По умолчанию — True. |
| Возвращает | |
|---|---|
Локальный tf.distribute.Server. |
join
join()
Блокирует выполнение до завершения работы сервера.
В настоящее время этот метод блокирует выполнение навсегда.
| Вызывает исключения | |
|---|---|
tf.errors.OpError | Или один из его подклассов, если при подключении к серверу TensorFlow произошла ошибка. |
start
start()
Запускает этот сервер.
| Вызывает исключения | |
|---|---|
tf.errors.OpError | Или один из его подклассов, если при запуске сервера TensorFlow произошла ошибка. |
© 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/distribute/Server