LoadTPUEmbeddingMomentumParameters

LoadTPUEmbeddingMomentumParameters classe final públicas

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

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

Métodos Públicos

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

Métodos herdados

Métodos Públicos

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

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

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

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

public static LoadTPUEmbeddingMomentumParameters.Options tableId (Long tableId)

public static LoadTPUEmbeddingMomentumParameters.Options tableName (String tableName)