RetrieveTPUEmbeddingADAMParametersGradAccumDebug

clase final pública RetrieveTPUEmbeddingADAMParametersGradAccumDebug

Recupere los parámetros de incrustación de ADAM con soporte de depuración.

Una operación que recupera los parámetros de optimización de la incrustación en la memoria del host. 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 utiliza para recuperar parámetros actualizados antes de guardar un punto de control.

Clases anidadas

clase RetrieveTPUEmbeddingADAMParametersGradAccumDebug.Options Atributos opcionales para RetrieveTPUEmbeddingADAMParametersGradAccumDebug

Métodos públicos

static RetrieveTPUEmbeddingADAMParametersGradAccumDebug.Options
static RetrieveTPUEmbeddingADAMParametersGradAccumDebug
crear ( alcance de alcance, fragmentos numéricos largos, ID de fragmento largo, opciones ... opciones)
Método de fábrica para crear una clase que envuelva una nueva operación RetrieveTPUEmbeddingADAMParametersGradAccumDebug.
Salida <Float>
Gradiente Acumuladores ()
Parámetro gradiente_acumuladores actualizado por el algoritmo de optimización de ADAM.
Salida <Float>
moment ()
Momentos de parámetros actualizados por el algoritmo de optimización de ADAM.
Salida <Float>
parámetros ()
Parámetros de parámetros actualizados por el algoritmo de optimización de ADAM.
static RetrieveTPUEmbeddingADAMParametersGradAccumDebug.Options
tableId (Long tableId)
static RetrieveTPUEmbeddingADAMParametersGradAccumDebug.Options
tableName (cadena tableName)
Salida <Float>
velocidades ()
Velocidades de parámetros actualizadas por el algoritmo de optimización de ADAM.

Métodos heredados

Métodos públicos

Public static RetrieveTPUEmbeddingADAMParametersGradAccumDebug.Options config (String config)

public static RetrieveTPUEmbeddingADAMParametersGradAccumDebug create ( Alcance del alcance, Long numShards, Long shardId, Opciones ... opciones)

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

Parámetros
alcance alcance actual
opciones lleva valores de atributos opcionales
Devoluciones
  • una nueva instancia de RetrieveTPUEmbeddingADAMParametersGradAccumDebug

Public Output <Float> gradientAccumulators ()

Parámetro gradiente_acumuladores actualizado por el algoritmo de optimización de ADAM.

Salida pública <Float> momenta ()

Momentos de parámetros actualizados por el algoritmo de optimización de ADAM.

parámetros de salida pública <Float> ()

Parámetros de parámetros actualizados por el algoritmo de optimización de ADAM.

public static RetrieveTPUEmbeddingADAMParametersGradAccumDebug.Options tableId (Long tableId)

public static RetrieveTPUEmbeddingADAMParametersGradAccumDebug.Options tableName (String tableName)

pública Salida <> Float velocidades ()

Velocidades de parámetros actualizadas por el algoritmo de optimización de ADAM.