Cargue los parámetros de incrustación de ADAM.
Una operación que carga parámetros de optimización en HBM para incrustarlos. Debe ir precedido de una operación ConfigureTPUEmbeddingHost que establezca la configuración correcta de la tabla de inserción. Por ejemplo, esta operación se usa para instalar parámetros que se cargan desde un punto de control antes de que se ejecute un ciclo de entrenamiento.
Clases anidadas
clase | LoadTPUEmbeddingADAMParameters.Options | Los atributos opcionales para LoadTPUEmbeddingADAMParameters |
Constantes
Cuerda | OP_NAME | El nombre de esta operación, como lo conoce el motor central de TensorFlow |
Métodos públicos
estáticas LoadTPUEmbeddingADAMParameters.Options | config (String config) |
estáticas LoadTPUEmbeddingADAMParameters | |
estáticas LoadTPUEmbeddingADAMParameters.Options | tableId no (Long tableId no) |
estáticas LoadTPUEmbeddingADAMParameters.Options | nombreTabla (String nombreTabla) |
Métodos heredados
Constantes
OP_NAME pública final static String
El nombre de esta operación, como lo conoce el motor central de TensorFlow
Métodos públicos
estáticas pública LoadTPUEmbeddingADAMParameters crean ( Alcance alcance, operando < TFloat32 > parámetros, operando < TFloat32 > momentos, operando < TFloat32 > velocidades, Long numShards, Long shardId, Opciones ... Opciones)
Método de fábrica para crear una clase que envuelva una nueva operación LoadTPUEmbeddingADAMParameters.
Parámetros
alcance | alcance actual |
---|---|
parámetros | Valor de los parámetros utilizados en el algoritmo de optimización de ADAM. |
Momentos | Valor de los momentos utilizados en el algoritmo de optimización de ADAM. |
velocidades | Valor de las velocidades utilizadas en el algoritmo de optimización de ADAM. |
opciones | lleva valores de atributos opcionales |
Devoluciones
- una nueva instancia de LoadTPUEmbeddingADAMParameters