LoadTPUEmbeddingADAMParametersGradAccumDebug

clase final pública LoadTPUEmbeddingADAMParametersGradAccumDebug

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

Una operación que carga parámetros de optimización en HBM para su integración. Debe estar precedido por una operación ConfigureTPUEmbeddingHost que establezca la configuración correcta de la tabla de incrustación. Por ejemplo, esta operación se utiliza para instalar parámetros que se cargan desde un punto de control antes de que se ejecute un ciclo de entrenamiento.

Clases anidadas

clase CargarTPUEmbeddingADAMParametersGradAccumDebug.Options Atributos opcionales para LoadTPUEmbeddingADAMParametersGradAccumDebug

Métodos públicos

Carga estáticaTPUEmbeddingADAMParametersGradAccumDebug.Options
config (configuración de cadena)
Carga estáticaTPUEmbeddingADAMParametersGradAccumDebug
crear (alcance alcance , parámetros del operando <Float>, momentos del operando <Float>, velocidades del operando <Float>, acumuladores de gradiente del operando <Float>, Long numShards, Long shardId, Opciones... opciones)
Método de fábrica para crear una clase que envuelve una nueva operación LoadTPUEmbeddingADAMParametersGradAccumDebug.
Carga estáticaTPUEmbeddingADAMParametersGradAccumDebug.Options
tableId (Id de tabla larga)
Carga estáticaTPUEmbeddingADAMParametersGradAccumDebug.Options
nombre de tabla (nombre de tabla de cadena)

Métodos heredados

Métodos públicos

Configuración pública estática LoadTPUEmbeddingADAMParametersGradAccumDebug.Options (configuración de cadena)

LoadTPUEmbeddingADAMParametersGradAccumDebug estático público crear (alcance del alcance , parámetros del operando <Float>, momentos del operando <Float>, velocidades del operando <Float>, acumuladores de gradiente del operando <Float>, números largos de fragmentos, ID de fragmento largo, opciones... opciones)

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

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

LoadTPUEmbeddingADAMParametersGradAccumDebug.Options tableId (ID de tabla larga) estático público

LoadTPUEmbeddingADAMParametersGradAccumDebug.Options tableName (cadena nombre de tabla) estático público