LoadTPUEmbeddingStochasticGradientDescentParameters

public final class LoadTPUEmbeddingStochasticGradientDescentParameters

Cargue los parámetros de incrustación de SGD.

Una operación que carga parámetros de optimización en HBM para incrustarlos. Debe ir precedido de una operación ConfigureTPUEmbeddingHost que establezca la configuración correcta de la tabla de inserción. Por ejemplo, esta operación se usa para instalar parámetros que se cargan desde un punto de control antes de que se ejecute un ciclo de entrenamiento.

Clases anidadas

clase LoadTPUEmbeddingStochasticGradientDescentParameters.Options Atributos opcionales para LoadTPUEmbeddingStochasticGradientDescentParameters

Métodos públicos

static LoadTPUEmbeddingStochasticGradientDescentParameters.Options
static LoadTPUEmbeddingStochasticGradientDescentParameters
create ( alcance del alcance, parámetros del operando <Float>, Long numShards, Long shardId, opciones ... opciones)
Método de fábrica para crear una clase que envuelva una nueva operación LoadTPUEmbeddingStochasticGradientDescentParameters.
static LoadTPUEmbeddingStochasticGradientDescentParameters.Options
tableId (Long tableId)
static LoadTPUEmbeddingStochasticGradientDescentParameters.Options
tableName (cadena tableName)

Métodos heredados

Métodos públicos

public static LoadTPUEmbeddingStochasticGradientDescentParameters create ( Alcance del alcance, Operand <Float> parámetros, Long numShards, Long shardId, Opciones ... opciones)

Método de fábrica para crear una clase que envuelva una nueva operación LoadTPUEmbeddingStochasticGradientDescentParameters.

Parámetros
alcance alcance actual
parámetros Valor de los parámetros utilizados en el algoritmo de optimización del descenso de gradiente estocástico.
opciones lleva valores de atributos opcionales
Devoluciones
  • una nueva instancia de LoadTPUEmbeddingStochasticGradientDescentParameters

public static LoadTPUEmbeddingStochasticGradientDescentParameters.Options tableName (String tableName)