LoadTPUEmbeddingMomentumParametersGradAccumDebug

публичный финальный класс LoadTPUEmbeddingMomentumParametersGradAccumDebug

Загрузка параметров внедрения Momentum с поддержкой отладки.

Операция, которая загружает параметры оптимизации в HBM для встраивания. Должна предшествовать операция ConfigureTPUEmbeddingHost, которая устанавливает правильную конфигурацию таблицы внедрения. Например, эта операция используется для установки параметров, которые загружаются из контрольной точки перед выполнением цикла обучения.

Вложенные классы

сорт LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options Дополнительные атрибуты для LoadTPUEmbeddingMomentumParametersGradAccumDebug

Публичные методы

статический LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options
конфигурация (строковая конфигурация)
статическая нагрузкаTPUEmbeddingMomentumParametersGradAccumDebug
create ( Область области, параметры операнда <Float>, импульсы операнда <Float>, градиентные аккумуляторы операнда <Float>, длинные numShards, длинные shardId, параметры... )
Фабричный метод для создания класса, обертывающего новую операцию LoadTPUEmbeddingMomentumParametersGradAccumDebug.
статический LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options
tableId (длинный tableId)
статический LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options
tableName (строка tableName)

Унаследованные методы

Публичные методы

общедоступная статическая конфигурация LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options (конфигурация String)

public static LoadTPUEmbeddingMomentumParametersGradAccumDebug create (область области действия , параметры операнда <Float>, моменты операнда <Float>, операнд <Float> градиентаАккумуляторы, длинные numShards, длинные shardId, параметры... параметры)

Фабричный метод для создания класса, обертывающего новую операцию LoadTPUEmbeddingMomentumParametersGradAccumDebug.

Параметры
объем текущий объем
параметры Значение параметров, используемых в алгоритме оптимизации Momentum.
импульс Значение импульсов, используемых в алгоритме оптимизации Momentum.
ГрадиентАккумуляторы Значение градиента_аккумуляторов, используемое в алгоритме оптимизации Momentum.
параметры содержит значения необязательных атрибутов
Возврат
  • новый экземпляр LoadTPUEmbeddingMomentumParametersGradAccumDebug

общедоступный статический LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options tableId (Long tableId)

public static LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options tableName (String tableName)