LoadTPUEmbeddingMomentumParametersGradAccumDebug

LoadTPUEmbeddingMomentumParametersGradAccumDebug classe final pública

Carregue os parâmetros de incorporação do Momentum com suporte para depuração.

Uma operação que carrega parâmetros de otimização no HBM para incorporação. Deve ser precedido por um op ConfigureTPUEmbeddingHost que define a configuração correta da tabela de incorporação. Por exemplo, esta operação é usada para instalar parâmetros que são carregados de um ponto de verificação antes que um loop de treinamento seja executado.

Classes aninhadas

classe LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options Atributos opcionais para LoadTPUEmbeddingMomentumParametersGradAccumDebug

Métodos Públicos

estáticos LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options
configuração (String config)
estática LoadTPUEmbeddingMomentumParametersGradAccumDebug
criar ( Scope escopo, Operando <float> Parâmetros, Operando <float> Momenta, Operando <float> gradientAccumulators, numShards longo, longo shardId, Options ... Opções)
Método de fábrica para criar uma classe que envolve uma nova operação LoadTPUEmbeddingMomentumParametersGradAccumDebug.
estáticos LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options
tableId (Long tableId)
estáticos LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options
tableName (String tableName)

Métodos herdados

Métodos Públicos

public static LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options de configuração (config String)

public static LoadTPUEmbeddingMomentumParametersGradAccumDebug criar ( Scope escopo, Operando <float> Parâmetros, Operando <float> Momenta, Operando <float> gradientAccumulators, numShards longo, longo shardId, Options ... Opções)

Método de fábrica para criar uma classe que envolve uma nova operação LoadTPUEmbeddingMomentumParametersGradAccumDebug.

Parâmetros
alcance escopo atual
parametros Valor dos parâmetros usados ​​no algoritmo de otimização do Momentum.
momento Valor dos momentos usados ​​no algoritmo de otimização do momento.
gradiente Acumuladores Valor de gradiente_acumuladores usados ​​no algoritmo de otimização de momento.
opções carrega valores de atributos opcionais
Devoluções
  • uma nova instância de LoadTPUEmbeddingMomentumParametersGradAccumDebug

public static LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options tableId (Long tableId)

public static LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options tableName (String tableName)