LoadTPUEmbeddingRMSPropParametersGradAccumDebug

clase final pública LoadTPUEmbeddingRMSPropParametersGradAccumDebug

Cargue los parámetros de incrustación de RMSProp 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 LoadTPUEmbeddingRMSPropParametersGradAccumDebug.Options Atributos opcionales para LoadTPUEmbeddingRMSPropParametersGradAccumDebug

Métodos públicos

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

Métodos heredados

Métodos públicos

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

LoadTPUEmbeddingRMSPropParametersGradAccumDebug estático público crear (alcance del alcance , parámetros del operando <Float>, operando <Float> ms, operando <Float> mom, operando <Float> gradientAccumulators, Long numShards, Long shardId, Opciones... opciones)

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

Parámetros
alcance alcance actual
parámetros Valor de los parámetros utilizados en el algoritmo de optimización RMSProp.
EM Valor de ms utilizado en el algoritmo de optimización RMSProp.
mamá Valor de mamá utilizado en el algoritmo de optimización RMSProp.
gradienteAcumuladores Valor de gradient_accumuladores utilizados en el algoritmo de optimización RMSProp.
opciones lleva valores de atributos opcionales
Devoluciones
  • una nueva instancia de LoadTPUEmbeddingRMSPropParametersGradAccumDebug

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

LoadTPUEmbeddingRMSPropParametersGradAccumDebug.Options tableName (cadena tableName) estático público