LoadTPUEmbeddingCenteredRMSPropParameters

clase final pública LoadTPUEmbeddingCenteredRMSPropParameters

Parámetros de incrustación RMSProp centrados en la carga.

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 LoadTPUEmbeddingCenteredRMSPropParameters.Options Atributos opcionales para LoadTPUEmbeddingCenteredRMSPropParameters

Métodos públicos

static LoadTPUEmbeddingCenteredRMSPropParameters.Options
static LoadTPUEmbeddingCenteredRMSPropParameters
create ( Scope scope, Operand <Float> parámetros, Operand <Float> ms, Operand <Float> mom, Operand <Float> mg, Long numShards, Long shardId, Opciones ... opciones)
Método de fábrica para crear una clase que envuelva una nueva operación LoadTPUEmbeddingCenteredRMSPropParameters.
static LoadTPUEmbeddingCenteredRMSPropParameters.Options
tableId (Long tableId)
static LoadTPUEmbeddingCenteredRMSPropParameters.Options
tableName (cadena tableName)

Métodos heredados

Métodos públicos

public static LoadTPUEmbeddingCenteredRMSPropParameters create ( alcance de alcance, operando <Float> parámetros, operando <Float> ms, operando <Float> mom, operando <Float> mg, Long numShards, Long shardId, Opciones ... opciones)

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

Parámetros
alcance alcance actual
parámetros Valor de los parámetros utilizados en el algoritmo de optimización RMSProp centrado.
Sra Valor de ms utilizado en el algoritmo de optimización RMSProp centrado.
mamá Valor de mamá utilizado en el algoritmo de optimización RMSProp centrado.
mg Valor de mg utilizado en el algoritmo de optimización RMSProp centrado.
opciones lleva valores de atributos opcionales
Devoluciones
  • una nueva instancia de LoadTPUEmbeddingCenteredRMSPropParameters

public static LoadTPUEmbeddingCenteredRMSPropParameters.Options tableId (Long tableId)

public static LoadTPUEmbeddingCenteredRMSPropParameters.Options tableName (String tableName)