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

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

Модули

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

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

rpc модуль: Общедоступный 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: Синхронное обучение на TPUs и TPU-узлах.

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

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