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