LoadTPUEmbeddingADAMParameters

classe final pública 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 uma operação ConfigureTPUEmbeddingHost que define a configuração correta da tabela de incorporação. Por exemplo, este op é usado para instalar parâmetros que são carregados de um checkpoint antes que um loop de treinamento seja executado.

Classes aninhadas

aula LoadTPUEembeddingADAMParameters.Options Atributos opcionais para LoadTPUEmbeddingADAMParameters

Métodos públicos

static LoadTPUEmbeddingADAMParameters.Options
config (configuração de string)
static LoadTPUEembeddingADAMParameters
create ( Escopo do escopo, Parâmetros <Float> do Operando, Momento do Operando <Float>, Velocidades do Operando < Float >, Long numShards, Long shardId, Opções... opções)
Método de fábrica para criar uma classe que envolve uma nova operação LoadTPUEmbeddingADAMParameters.
static LoadTPUEmbeddingADAMParameters.Options
tableId (tabela longa)
static LoadTPUEmbeddingADAMParameters.Options
tableName (String tableName)

Métodos Herdados

Métodos públicos

configuração pública estática LoadTPUEmbeddingADAMParameters.Options ( configuração de string)

public static LoadTPUEmbeddingADAMParameters create ( Escopo do escopo, Parâmetros do Operando <Float>, Momentos do Operando <Float>, Velocidades do Operando < Float>, Long numShards, Long shardId, Opções... opções)

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

Parâmetros
alcance escopo atual
parâmetros Valor dos parâmetros usados ​​no algoritmo de otimização do ADAM.
momento Valor dos momentos usados ​​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

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

public static LoadTPUEmbeddingADAMParameters.Options tableName (String tableName)