LoadTPUEmbeddingADAMParameters

public final class LoadTPUEmbeddingADAMParameters

Carregue os parâmetros de incorporação do ADAM.

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 LoadTPUEmbeddingADAMParameters.Options Atributos opcionais para LoadTPUEmbeddingADAMParameters

Métodos Públicos

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

Métodos herdados

Métodos Públicos

public static LoadTPUEmbeddingADAMParameters.Options config (String config)

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

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

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.
opções carrega valores de atributos opcionais
Devoluções
  • uma nova instância de LoadTPUEmbeddingADAMParameters

public static LoadTPUEmbeddingADAMParameters.Options tableId (Long tableId)

public static LoadTPUEmbeddingADAMParameters.Options tableName (String tableName)