LoadTPUEmbeddingMomentumParametersGradAccumDebug

LoadTPUEmbeddingMomentumParametersGradAccumDebug public final class

Cargue los parámetros de incrustación de Momentum con soporte de depuración.

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 LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options Los atributos opcionales para LoadTPUEmbeddingMomentumParametersGradAccumDebug

Métodos públicos

estáticas LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options
config (String config)
estática LoadTPUEmbeddingMomentumParametersGradAccumDebug
crear ( Alcance alcance, operando <> Float parámetros, operando <> Float Momenta, operando <> Float gradientAccumulators, numShards mucho, mucho shardId, Opciones ... Opciones)
Método de fábrica para crear una clase que envuelva una nueva operación LoadTPUEmbeddingMomentumParametersGradAccumDebug.
estáticas LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options
tableId no (Long tableId no)
estáticas LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options
nombreTabla (String nombreTabla)

Métodos heredados

Métodos públicos

estáticas pública LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options de configuración (config String)

public static LoadTPUEmbeddingMomentumParametersGradAccumDebug crear ( Alcance alcance, operando <> Float parámetros, operando <> Float Momenta, operando <> Float gradientAccumulators, numShards mucho, mucho shardId, Opciones ... Opciones)

Método de fábrica para crear una clase que envuelva una nueva operación LoadTPUEmbeddingMomentumParametersGradAccumDebug.

Parámetros
alcance alcance actual
parámetros Valor de los parámetros utilizados en el algoritmo de optimización Momentum.
Momentos Valor de los momentos utilizados en el algoritmo de optimización del momento.
gradiente Acumuladores Valor de los acumuladores de gradientes utilizados en el algoritmo de optimización de Momentum.
opciones lleva valores de atributos opcionales
Devoluciones
  • una nueva instancia de LoadTPUEmbeddingMomentumParametersGradAccumDebug

public static LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options tableId no (Long tableId no)

public static LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options nombreTabla (String nombreTabla)