Модуль: tf.distribute.experimental

Библиотека экспериментальных стратегий распределения.

Модули

coordinator модуль: Общедоступный API для пространства имён tf.distribute.experimental.coordinator.

partitioners модуль: Общедоступный API для пространства имён tf.distribute.experimental.partitioners.

rpc модуль: Общедоступный API для пространства имён tf.distribute.experimental.rpc.

Классы

class CentralStorageStrategy: Стратегия для одной машины, размещающая все переменные на одном устройстве.

class CollectiveCommunication: Реализация меж-устройства коммуникаций.

class CollectiveHints: Указания для коллективных операций, таких как AllReduce.

class CommunicationImplementation: Реализация меж-устройства коммуникаций.

class CommunicationOptions: Параметры для меж-устройства коммуникаций, таких как All-reduce.

class MultiWorkerMirroredStrategy: Стратегия распределения для синхронного обучения на нескольких рабочих узлах.

class ParameterServerStrategy: Стратегия tf.distribute для многоузлового обучения с параметровыми серверами.

class TPUStrategy: Синхронное обучение на TPU и TPU-кластерах.

class ValueContext: Класс, содержащий информацию, необходимую для распределенной функции.