LoadTPUEmbeddingMomentumParametersGradAccumDebug

clase final pública LoadTPUEmbeddingMomentumParametersGradAccumDebug

Cargue los parámetros de incorporación de Momentum 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 LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options Atributos opcionales para LoadTPUEmbeddingMomentumParametersGradAccumDebug

Métodos públicos

Carga estáticaTPUEmbeddingMomentumParametersGradAccumDebug.Options
config (configuración de cadena)
Carga estáticaTPUEmbeddingMomentumParametersGradAccumDebug
crear (alcance alcance , parámetros del operando <Float>, momentos 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 LoadTPUEmbeddingMomentumParametersGradAccumDebug.
Carga estáticaTPUEmbeddingMomentumParametersGradAccumDebug.Options
tableId (Id de tabla larga)
Carga estáticaTPUEmbeddingMomentumParametersGradAccumDebug.Options
nombre de tabla (nombre de tabla de cadena)

Métodos heredados

Métodos públicos

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

creación pública estática LoadTPUEmbeddingMomentumParametersGradAccumDebug (alcance del alcance , parámetros del operando <Float>, momentos 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 LoadTPUEmbeddingMomentumParametersGradAccumDebug.

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

LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options tableId estático público (Long tableId)

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