LoadTPUEmbeddingADAMParameters

clase final pública LoadTPUEmbeddingADAMParameters

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

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

Métodos públicos

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

Métodos heredados

Métodos públicos

Public static LoadTPUEmbeddingADAMParameters.Options config (String config)

Public static LoadTPUEmbeddingADAMParameters create ( Scope scope, Operand <Float> parámetros, Operand <Float> momentos, Operand <Float> velocidades, Long numShards, Long shardId, Opciones ... opciones)

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

Parámetros
alcance alcance actual
parámetros Valor de los parámetros utilizados en el algoritmo de optimización de ADAM.
Momentos Valor de los momentos utilizados en el algoritmo de optimización de ADAM.
velocidades Valor de las velocidades utilizadas en el algoritmo de optimización de ADAM.
opciones lleva valores de atributos opcionales
Devoluciones
  • una nueva instancia de LoadTPUEmbeddingADAMParameters

public static LoadTPUEmbeddingADAMParameters.Options tableId (Long tableId)

Public static LoadTPUEmbeddingADAMParameters.Options tableName (String tableName)