tf.contrib.tpu.TPUDistributionStrategy

Стратегия запуска модели Keras на TPU.

tf.contrib.tpu.TPUDistributionStrategy(
    tpu_cluster_resolver=None, using_single_core=False
)

Аргументы

tpu_cluster_resolver Любой экземпляр TPUClusterResolver. Если None, будет создан экземпляр с '' в качестве адреса master.

using_single_core Булево значение. Это опция отладки, которая может быть удалена в будущем, после того, как функциональность репликации модели станет достаточно зрелой. Если False (по умолчанию), система автоматически находит наилучшую конфигурацию, с точки зрения количества ядер TPU, для репликации модели, как правило, используя все доступные ядра TPU. Если переопределяется как True, принудительно используется репликация модели с одним ядром, т.е. без репликации.

Аргументы
`tpu_cluster_resolver`	Любой экземпляр `TPUClusterResolver`. Если None, будет создан экземпляр с '' в качестве адреса master.
`using_single_core`	Булево значение. Это опция отладки, которая может быть удалена в будущем, после того, как функциональность репликации модели станет достаточно зрелой. Если `False` (по умолчанию), система автоматически находит наилучшую конфигурацию, с точки зрения количества ядер TPU, для репликации модели, как правило, используя все доступные ядра TPU. Если переопределяется как `True`, принудительно используется репликация модели с одним ядром, т.е. без репликации.

Исключения
`Exception`	На данном работнике не найден TPU.

© 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/r1.15/api_docs/python/tf/contrib/tpu/TPUDistributionStrategy