tf.raw_ops.LoadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug

Загрузка параметров SGD вложения.

Просмотр псевдонимов

Псевдонимы для миграции

См. Руководство по миграции для получения дополнительной информации.

tf.compat.v1.raw_ops.LoadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug

tf.raw_ops.LoadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug(
    parameters, gradient_accumulators, num_shards, shard_id, table_id=-1,
    table_name='', config='', name=None
)

Операция, которая загружает параметры оптимизации в HBM для вложения. Должна быть предшествующей операцией ConfigureTPUEmbeddingHost, которая настраивает правильную конфигурацию таблицы вложений. Например, эта операция используется для установки параметров, загруженных из контрольной точки, перед выполнением цикла обучения.

Аргументы
`parameters`	Массив типа `float32`. Значения параметров, используемых в алгоритме оптимизации стохастического градиентного спуска.
`gradient_accumulators`	Массив типа `float32`. Значения накопителей градиента, используемых в алгоритме оптимизации Adadelta.
`num_shards`	Целое число.
`shard_id`	Целое число.
`table_id`	Необязательное целое число. По умолчанию `-1`.
`table_name`	Необязательное целое число. По умолчанию `""`.
`config`	Необязательное целое число. По умолчанию `""`.
`name`	Имя операции (необязательно).

Возвращаемые значения
Созданная операция.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/raw_ops/LoadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug