tf.raw_ops.LoadTPUEmbeddingFTRLParameters
Загрузка параметров эмбеддинга FTRL.
tf.raw_ops.LoadTPUEmbeddingFTRLParameters(
parameters,
accumulators,
linears,
num_shards,
shard_id,
table_id=-1,
table_name='',
config='',
name=None
)
Операция, загружающая параметры оптимизации в HBM для эмбеддинга. Перед её выполнением необходимо выполнить операцию ConfigureTPUEmbeddingHost, которая настраивает соответствующую конфигурацию таблицы эмбеддингов. Например, эту операцию используют для установки параметров, загруженных из контрольной точки, перед запуском цикла обучения.
| Аргументы | |
|---|---|
parameters | Tensor типа float32. Значение параметров, используемых в алгоритме оптимизации FTRL. |
accumulators | Tensor типа float32. Значение аккумуляторов, используемых в алгоритме оптимизации FTRL. |
linears | Tensor типа float32. Значение линейных параметров, используемых в алгоритме оптимизации FTRL. |
num_shards | int. |
shard_id | int. |
table_id | Необязательный int. Значение по умолчанию: -1. |
table_name | Необязательный string. Значение по умолчанию: "". |
config | Необязательный string. Значение по умолчанию: "". |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
| Созданная операция. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/raw_ops/LoadTPUEmbeddingFTRLParameters