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