LoadTPUEmbeddingADAMParametersGradAccumDebug

public final class LoadTPUEmbeddingADAMParametersGradAccumDebug

Carregue os parâmetros de incorporação do ADAM 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

aula LoadTPUEmbeddingADAMParametersGradAccumDebug.Options Atributos opcionais para LoadTPUEmbeddingADAMParametersGradAccumDebug

Métodos Públicos

static LoadTPUEmbeddingADAMParametersGradAccumDebug.Options
static LoadTPUEmbeddingADAMParametersGradAccumDebug
criar ( Scope escopo, Operando <float> Parâmetros, Operando <float> Momenta, Operando <float> velocidades, 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 LoadTPUEmbeddingADAMParametersGradAccumDebug.
static LoadTPUEmbeddingADAMParametersGradAccumDebug.Options
tableId (Long tableId)
static LoadTPUEmbeddingADAMParametersGradAccumDebug.Options
tableName (String tableName)

Métodos herdados

Métodos Públicos

public static LoadTPUEmbeddingADAMParametersGradAccumDebug.Options config (String config)

public static LoadTPUEmbeddingADAMParametersGradAccumDebug criar ( Scope escopo, Operando <float> Parâmetros, Operando <float> Momenta, Operando <float> velocidades, Operando <float> gradientAccumulators, Longo numShards, Long shardId, Opções ... Opções)

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

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

public static LoadTPUEmbeddingADAMParametersGradAccumDebug.Options tableId (Long tableId)

public static LoadTPUEmbeddingADAMParametersGradAccumDebug.Options tableName (String tableName)