LoadTPUEmbeddingCenteredRMSPropParameters

LoadTPUEmbeddingCenteredRMSPropParameters classe final públicas

Carregar parâmetros de incorporação RMSProp centralizados.

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

Métodos Públicos

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

Métodos herdados

Métodos Públicos

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

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

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

Parâmetros
alcance escopo atual
parametros Valor dos parâmetros usados ​​no algoritmo de otimização RMSProp centralizado.
em Valor de ms usado no algoritmo de otimização RMSProp centralizado.
mãe Valor de mom usado no algoritmo de otimização RMSProp centralizado.
mg Valor de mg usado no algoritmo de otimização RMSProp centralizado.
opções carrega valores de atributos opcionais
Devoluções
  • uma nova instância de LoadTPUEmbeddingCenteredRMSPropParameters

public static LoadTPUEmbeddingCenteredRMSPropParameters.Options tableId (Long tableId)

public static LoadTPUEmbeddingCenteredRMSPropParameters.Options tableName (String tableName)