Uma operação que carrega parâmetros de otimização na memória incorporada.
Uma operação que carrega parâmetros de otimização na memória incorporada. 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. Para Adagrad, auxiliares1 devem ser os acumuladores. Para SGD, todos os valores auxiliares* devem estar vazios. Para FTRL, auxiliar1 deve ser os acumuladores e auxiliar2 deve ser os termos lineares. Para ADAM, auxiliar1 deve ser os momentos e auxiliar2 devem ser as velocidades.
Métodos públicos
Parâmetros estáticos LoadAllTPUEmbedding | create ( Escopo do escopo, Iterable< Operando <Float>> parâmetros, Iterable< Operando <Float>> auxiliar1, Iterable< Operando <Float>> auxiliar2, Iterable< Operando <Float>> auxiliar3, Iterable< Operando <Float>> auxiliary4, Iterable< Operando <Float>> auxiliar5, Iterable< Operando <Float>> auxiliar6, Iterable< Operando <Float>> auxiliary7, String config, Long numShards, Long shardId) Método de fábrica para criar uma classe que envolve uma nova operação LoadAllTPUEmbeddingParameters. |
Métodos Herdados
Métodos públicos
public static LoadAllTPUEmbeddingParameters create ( Escopo do escopo, Iterable< Operando <Float>> parâmetros, Iterable< Operando <Float>> auxiliar1, Iterable< Operando <Float>> auxiliar2, Iterable< Operando <Float>> auxiliar3, Iterable< Operando <Float> > auxiliar4, Iterable< Operando <Float>> auxiliar5, Iterable< Operando <Float>> auxiliar6, Iterable< Operando <Float>> auxiliary7, String config, Long numShards, Long shardId)
Método de fábrica para criar uma classe que envolve uma nova operação LoadAllTPUEmbeddingParameters.
Parâmetros
alcance | escopo atual |
---|---|
parâmetros | Uma lista de tensores, um para cada tabela de incorporação, contendo os parâmetros iniciais da tabela de incorporação a serem usados em pesquisas de incorporação. |
auxiliar1 | Uma lista de tensores, um para cada tabela de incorporação, contendo os valores iniciais do primeiro parâmetro de otimização auxiliar a ser usado nas atualizações de loop de treinamento de incorporação. A forma de cada entrada é ignorada (e, portanto, pode estar vazia) para aquelas tabelas cujos algoritmos de otimização não possuem pelo menos um parâmetro auxiliar. |
auxiliar2 | Uma lista de tensores, um para cada tabela de incorporação, contendo os valores iniciais do segundo parâmetro de otimização auxiliar a ser usado nas atualizações de loop de treinamento de incorporação. A forma de cada entrada é ignorada (e, portanto, pode estar vazia) para aquelas tabelas cujos algoritmos de otimização não possuem pelo menos duas |
auxiliar3 | Uma lista de tensores, um para cada tabela de incorporação, contendo os valores iniciais do terceiro parâmetro de otimização auxiliar a ser usado nas atualizações de loop de treinamento de incorporação. A forma de cada entrada é ignorada (e, portanto, pode estar vazia) para aquelas tabelas cujos algoritmos de otimização não possuem três parâmetros auxiliares. |
auxiliar4 | Uma lista de tensores, um para cada tabela de incorporação, contendo os valores iniciais do segundo parâmetro de otimização auxiliar a ser usado nas atualizações de loop de treinamento de incorporação. A forma de cada entrada é ignorada (e, portanto, pode estar vazia) para aquelas tabelas cujos algoritmos de otimização não possuem pelo menos quatro |
auxiliar5 | Uma lista de tensores, um para cada tabela de incorporação, contendo os valores iniciais do terceiro parâmetro de otimização auxiliar a ser usado nas atualizações de loop de treinamento de incorporação. A forma de cada entrada é ignorada (e, portanto, pode estar vazia) para aquelas tabelas cujos algoritmos de otimização não possuem cinco parâmetros auxiliares. |
auxiliar6 | Uma lista de tensores, um para cada tabela de incorporação, contendo os valores iniciais do segundo parâmetro de otimização auxiliar a ser usado nas atualizações de loop de treinamento de incorporação. A forma de cada entrada é ignorada (e, portanto, pode estar vazia) para aquelas tabelas cujos algoritmos de otimização não possuem pelo menos seis |
auxiliar7 | Uma lista de tensores, um para cada tabela de incorporação, contendo os valores iniciais do terceiro parâmetro de otimização auxiliar a ser usado nas atualizações de loop de treinamento de incorporação. A forma de cada entrada é ignorada (e, portanto, pode estar vazia) para aquelas tabelas cujos algoritmos de otimização não possuem parâmetros auxiliares sevan. |
configuração | Um protocolo TPUEmbeddingConfiguration que descreve os parâmetros da tabela que estão sendo carregados, serializados em uma string. |
numShards | Número de estilhaços nos quais as tabelas de incorporação são divididas. |
shardId | Identificador do estilhaço para esta operação. |
Devoluções
- uma nova instância de LoadAllTPUEmbeddingParameters