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 |
Constantes
Cadena | OP_NOMBRE | El nombre de esta operación, como lo conoce el motor central de TensorFlow. |
Métodos públicos
Carga estáticaTPUEmbeddingMomentumParametersGradAccumDebug.Options | config (configuración de cadena) |
Carga estáticaTPUEmbeddingMomentumParametersGradAccumDebug | crear ( alcance alcance , parámetros de operando <TFloat32> , momentos de operando <TFloat32> , acumuladores de gradiente de operando <TFloat32> , numShards largos, Id de fragmentos largos, 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
Constantes
Cadena final estática pública OP_NAME
El nombre de esta operación, como lo conoce el motor central de TensorFlow.
Métodos públicos
Configuración pública estática LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options (configuración de cadena)
LoadTPUEmbeddingMomentumParametersGradAccumDebug estático público crear (alcance alcance , parámetros de operando <TFloat32> , momentos de operando <TFloat32> , acumuladores de gradiente de operando <TFloat32> , numShards largos, shardId largo, 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