tf.distribute.InputOptions

Параметры выполнения для experimental_distribute_dataset(s_from_function).

tf.distribute.InputOptions(
    experimental_prefetch_to_device=True,
    experimental_replication_mode=tf.distribute.InputReplicationMode.PER_WORKER,
    experimental_place_dataset_on_device=False
)

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

# Setup TPUStrategy
resolver = tf.distribute.cluster_resolver.TPUClusterResolver(tpu='')
tf.config.experimental_connect_to_cluster(resolver)
tf.tpu.experimental.initialize_tpu_system(resolver)
strategy = tf.distribute.TPUStrategy(resolver)

dataset = tf.data.Dataset.range(16)
distributed_dataset_on_host = (
    strategy.experimental_distribute_dataset(
        dataset,
        tf.distribute.InputOptions(
            experimental_replication_mode=
            experimental_replication_mode.PER_WORKER,
            experimental_place_dataset_on_device=False)))

Атрибуты
`experimental_prefetch_to_device`	Булево. По умолчанию True. Если True, элементы набора данных будут предварительно загружены в память устройства ускорителя. Если False, элементы набора данных предварительно загружаются в память устройства хоста. Должно быть False при использовании API TPUEmbedding. experimental_prefetch_to_device может быть использован только с experimental_replication_mode=PER_WORKER
`experimental_replication_mode`	Режим репликации для функции входных данных. В настоящее время режим InputReplicationMode.PER_REPLICA поддерживается только с tf.distribute.MirroredStrategy. experimental_distribute_datasets_from_function. Значение по умолчанию — InputReplicationMode.PER_WORKER.
`experimental_place_dataset_on_device`	Булево. По умолчанию False. Если True, набор данных будет размещён на устройстве, иначе он останется на хосте. experimental_place_dataset_on_device=True может быть использован только с experimental_replication_mode=PER_REPLICA

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/distribute/InputOptions