LoadTPUEmbeddingADAMParameters

clase final pública LoadTPUEmbeddingADAMParameters

Cargue los parámetros de incrustación de ADAM.

Una operación que carga parámetros de optimización en HBM para incrustarlos. Debe estar precedido por una operación ConfigureTPUEmbeddingHost que establece la configuración correcta de la tabla de incrustación. Por ejemplo, esta opció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 Atributos opcionales para LoadTPUEmbeddingADAMParameters

Métodos públicos

Carga estáticaTPUEmbeddingADAMParameters.Options
config (Configuración de cadena)
Carga estáticaTPUEmbeddingADAMParameters
create ( Scope scope, Operando <Flotante> parámetros, Operando <Flotante> momento, Operando <Flotante> velocidades, Long numShards, Long shardId, Opciones... opciones)
Método de fábrica para crear una clase que envuelve una nueva operación LoadTPUEmbeddingADAMParameters.
Carga estáticaTPUEmbeddingADAMParameters.Options
tableId (Id de tabla larga)
Carga estáticaTPUEmbeddingADAMParameters.Options
tableName (Cadena tableName)

Métodos Heredados

Métodos públicos

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

Public static LoadTPUEmbeddingADAMParameters create ( Scope scope, Operand <Float> meters, Operand <Float> momenta, Operand <Float> velocities, Long numShards, Long shardId, Options... options)

Método de fábrica para crear una clase que envuelve 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 momento utilizado 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

público estático LoadTPUEmbeddingADAMParameters.Options tableId (Long tableId)

público estático LoadTPUEmbeddingADAMParameters.Options tableName (String tableName)