Module: tf.distribute.experimental

Открытый API пространства имён tf._api.v2.distribute.experimental

Модули

coordinator module: Открытый API пространства имён tf._api.v2.distribute.experimental.coordinator

partitioners module: Открытый API пространства имён tf._api.v2.distribute.experimental.partitioners

rpc module: Открытый API пространства имён tf._api.v2.distribute.experimental.rpc

Классы

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

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

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

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

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

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

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

class PreemptionCheckpointHandler: Обработчик вытеснения и ошибок для синхронного обучения.

class PreemptionWatcher: Отслеживает сигнал вытеснения и сохраняет его.

class TPUStrategy: Синхронное обучение на TPU и модулях TPU.

class TerminationConfig: Настройка PreemptionCheckpointHandler для различных платформ.

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