LoadTPUEmbeddingMomentumParameters

Public final class LoadTPUEmbeddingMomentumParameters

Cargue los parámetros de incrustación de Momentum.

Una operación que carga parámetros de optimización en HBM para incrustarlos. Debe ir precedido de una operación ConfigureTPUEmbeddingHost que establezca la configuración correcta de la tabla de inserción. Por ejemplo, esta operación se usa para instalar parámetros que se cargan desde un punto de control antes de que se ejecute un ciclo de entrenamiento.

Clases anidadas

clase LoadTPUEmbeddingMomentumParameters.Options Atributos opcionales para LoadTPUEmbeddingMomentumParameters

Métodos públicos

static LoadTPUEmbeddingMomentumParameters.Options
Carga estática TPUEmbeddingMomentumParameters
crear ( Alcance alcance, operando <> Float parámetros, operando <> Float Momenta, numShards mucho, mucho shardId, Opciones ... Opciones)
Método de fábrica para crear una clase que envuelva una nueva operación LoadTPUEmbeddingMomentumParameters.
static LoadTPUEmbeddingMomentumParameters.Options
tableId (Long tableId)
static LoadTPUEmbeddingMomentumParameters.Options
tableName (cadena tableName)

Métodos heredados

Métodos públicos

Configuración pública estática LoadTPUEmbeddingMomentumParameters.Options (String config)

estáticas pública LoadTPUEmbeddingMomentumParameters crean ( Alcance alcance, operando <> Float parámetros, operando <> Float Momenta, numShards mucho, mucho shardId, Opciones ... Opciones)

Método de fábrica para crear una clase que envuelva una nueva operación LoadTPUEmbeddingMomentumParameters.

Parámetros
alcance alcance actual
parámetros Valor de los parámetros utilizados en el algoritmo de optimización Momentum.
Momentos Valor de los momentos utilizados en el algoritmo de optimización del momento.
opciones lleva valores de atributos opcionales
Devoluciones
  • una nueva instancia de LoadTPUEmbeddingMomentumParameters

public static LoadTPUEmbeddingMomentumParameters.Options tableId (Long tableId)

Public static LoadTPUEmbeddingMomentumParameters.Options tableName (String tableName)