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, 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. Para Adagrad, auxiliar1 devem ser os acumuladores. Para SGD, todos os valores auxiliares* devem estar vazios. Para FTRL, auxiliar1 devem ser os acumuladores e auxiliar2 devem ser os termos lineares. Para ADAM, auxiliar1 deve ser os momentos e auxiliar2 deve ser as velocidades.
Métodos Públicos
LoadAllTPUEmbeddingParameters estáticos | create (Escopo do escopo , Iterável< Operando <Float>> parâmetros, Iterável< Operando <Float>> auxiliar1, Iterável< Operando <Float>> auxiliar2, Iterável< Operando <Float>> auxiliar3, Iterável< Operando <Float>> auxiliar4, Iterável< Operando <Float>> auxiliar5, Iterável< Operando <Float>> auxiliar6, Iterável< Operando <Float>> auxiliar7, String config, Long numShards, Long shardId) Método de fábrica para criar uma classe que agrupa uma nova operação LoadAllTPUEmbeddingParameters. |
Métodos herdados
Métodos Públicos
public static LoadAllTPUEmbeddingParameters create ( Escopo escopo , Iterável< Operando <Float>> parâmetros, Iterável< Operando <Float>> auxiliar1, Iterável< Operando <Float>> auxiliar2, Iterável< Operando <Float>> auxiliar3, Iterável< Operando <Float> > auxiliar4, Iterável< Operando <Float>> auxiliar5, Iterável< Operando <Float>> auxiliar6, Iterável< Operando <Float>> auxiliar7, String config, Long numShards, Long shardId)
Método de fábrica para criar uma classe que agrupa uma nova operação LoadAllTPUEmbeddingParameters.
Parâmetros
escopo | 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 na incorporação de atualizações do loop de treinamento. 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 auxiliar de otimização a ser usado na incorporação de atualizações do loop de treinamento. A forma de cada entrada é ignorada (e, portanto, pode estar vazia) para aquelas tabelas cujos algoritmos de otimização não possuem pelo menos dois auxiliares. |
auxiliar3 | Uma lista de tensores, um para cada tabela de incorporação, contendo os valores iniciais do terceiro parâmetro auxiliar de otimização a ser usado na incorporação de atualizações do loop de treinamento. 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 auxiliar de otimização a ser usado na incorporação de atualizações do loop de treinamento. 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 auxiliares. |
auxiliar5 | Uma lista de tensores, um para cada tabela de incorporação, contendo os valores iniciais do terceiro parâmetro auxiliar de otimização a ser usado na incorporação de atualizações do loop de treinamento. 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 auxiliar de otimização a ser usado na incorporação de atualizações do loop de treinamento. 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 auxiliares. |
auxiliar7 | Uma lista de tensores, um para cada tabela de incorporação, contendo os valores iniciais do terceiro parâmetro auxiliar de otimização a ser usado na incorporação de atualizações do loop de treinamento. 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 proto TPUEmbeddingConfiguration que descreve os parâmetros da tabela que estão sendo carregados, serializados em uma string. |
numShards | Número de fragmentos nos quais as tabelas de incorporação são divididas. |
fragmentoId | Identificador do fragmento para esta operação. |
Devoluções
- uma nova instância de LoadAllTPUEmbeddingParameters