tf.distribute.InputContext

Класс, содержащий информацию, необходимую функции ввода.

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

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

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

tf.compat.v1.distribute.InputContext

tf.distribute.InputContext(
    num_input_pipelines=1, input_pipeline_id=0, num_replicas_in_sync=1
)

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

Аргументы
`num_input_pipelines`	количество конвейеров ввода в кластере.
`input_pipeline_id`	идентификатор текущего конвейера ввода; должен быть целым числом в диапазоне [0,`num_input_pipelines`).
`num_replicas_in_sync`	количество синхронизированных реплик.

Атрибуты
`input_pipeline_id`	Возвращает идентификатор конвейера ввода.
`num_input_pipelines`	Возвращает количество конвейеров ввода.
`num_replicas_in_sync`	Возвращает количество синхронизированных вычислительных реплик.

Методы

`get_per_replica_batch_size`

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

get_per_replica_batch_size(
    global_batch_size
)

Возвращает размер пакета для одной реплики.

Аргументы
`global_batch_size`	глобальный размер пакета, который должен делиться на `num_replicas_in_sync`.

Возвращает
размер пакета для одной реплики.

Вызывает исключение
`ValueError`	если `global_batch_size` не делится на `num_replicas_in_sync`.

© 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/InputContext