LoadTPUEmbeddingADAMParametersGradAccumDebug

LoadTPUEmbeddingADAMParametersGradAccumDebug public final class

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

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 LoadTPUEmbeddingADAMParametersGradAccumDebug.Options Los atributos opcionales para LoadTPUEmbeddingADAMParametersGradAccumDebug

Constantes

Cuerda OP_NAME El nombre de esta operación, como lo conoce el motor central de TensorFlow

Métodos públicos

estáticas LoadTPUEmbeddingADAMParametersGradAccumDebug.Options
config (String config)
estática LoadTPUEmbeddingADAMParametersGradAccumDebug
crear ( Alcance alcance, operando < TFloat32 > parámetros, operando < TFloat32 > momentos, operando < TFloat32 > velocidades, operando < TFloat32 > gradientAccumulators, Long numShards, Long shardId, Opciones ... Opciones)
Método de fábrica para crear una clase que envuelva una nueva operación LoadTPUEmbeddingADAMParametersGradAccumDebug.
estáticas LoadTPUEmbeddingADAMParametersGradAccumDebug.Options
tableId no (Long tableId no)
estáticas LoadTPUEmbeddingADAMParametersGradAccumDebug.Options
nombreTabla (String nombreTabla)

Métodos heredados

Constantes

OP_NAME pública final static String

El nombre de esta operación, como lo conoce el motor central de TensorFlow

Valor constante: "LoadTPUEmbeddingADAMParametersGradAccumDebug"

Métodos públicos

estáticas pública LoadTPUEmbeddingADAMParametersGradAccumDebug.Options de configuración (config String)

public static LoadTPUEmbeddingADAMParametersGradAccumDebug crear ( Alcance alcance, operando < TFloat32 > parámetros, operando < TFloat32 > momentos, operando < TFloat32 > velocidades, operando < TFloat32 > gradientAccumulators, Long numShards, Long shardId, Opciones ... Opciones)

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

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.
gradiente Acumuladores Valor de los acumuladores de gradientes utilizados en el algoritmo de optimización de ADAM.
opciones lleva valores de atributos opcionales
Devoluciones
  • una nueva instancia de LoadTPUEmbeddingADAMParametersGradAccumDebug

public static LoadTPUEmbeddingADAMParametersGradAccumDebug.Options tableId no (Long tableId no)

public static LoadTPUEmbeddingADAMParametersGradAccumDebug.Options nombreTabla (String nombreTabla)