tf.data.experimental.service.WorkerServer

Сервер ввода-вывода tf.data в процессе выполнения.

tf.data.experimental.service.WorkerServer(
    config, start=True
)

Объект tf.data.experimental.service.WorkerServer выполняет обработку tf.data.Dataset для настраиваемых пользователем наборов данных и предоставляет результирующие элементы через RPC. Рабочий процесс связан с одним tf.data.experimental.service.DispatchServer.

dispatcher = tf.data.experimental.service.DispatchServer()
dispatcher_address = dispatcher.target.split("://")[1]
worker = tf.data.experimental.service.WorkerServer(
    tf.data.experimental.service.WorkerConfig(
        dispatcher_address=dispatcher_address))
dataset = tf.data.Dataset.range(10)
dataset = dataset.apply(tf.data.experimental.service.distribute(
    processing_mode="parallel_epochs", service=dispatcher.target))
print(list(dataset.as_numpy_iterator()))
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

При запуске отдельного процесса tf.data worker, используйте join(), чтобы заблокировать выполнение после запуска worker, до завершения worker.

worker = tf.data.experimental.service.WorkerServer(
    port=5051, dispatcher_address="localhost:5050")
worker.join()

Вызовите stop(), чтобы корректно завершить работу worker. Worker автоматически завершается, когда все ссылки на него удалены.

Аргументы
`config`	Конфигурация `tf.data.experimental.service.WorkerConfig`.
`start`	(Необязательно.) Логическое значение, указывающее, следует ли запустить сервер после его создания. По умолчанию True.

Методы

`join`

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

join() -> None

Ожидает завершения работы сервера.

Это полезно при запуске отдельного процесса worker.

worker_server = tf.data.experimental.service.WorkerServer(
    port=5051, dispatcher_address="localhost:5050")
worker_server.join()

В настоящее время этот метод блокирует выполнение навсегда.

Исключения
`tf.errors.OpError`	Или одно из его подклассов, если произошла ошибка при присоединении к серверу.

`start`

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

start() -> None

Запускает этот сервер.

Исключения
`tf.errors.OpError`	Или одно из его подклассов, если произошла ошибка при запуске сервера.

`stop`

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

stop() -> None

Останавливает сервер.

Исключения
`tf.errors.OpError`	Или одно из его подклассов, если произошла ошибка при остановке сервера.

© 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/data/experimental/service/WorkerServer