tf.ConditionalAccumulatorBase
Условный сумматор для агрегирования градиентов.
tf.ConditionalAccumulatorBase(
dtype, shape, accumulator_ref
)
Градиенты, актуальные на текущий момент (т. е. момент вычисления градиента совпадает с текущим моментом сумматора), добавляются в сумматор.
Извлечение среднего градиента блокируется до тех пор, пока не будет накоплено необходимое количество градиентов.
| Аргументы | |
|---|---|
dtype | Тип данных накапливаемых градиентов. |
shape | Формат накапливаемых градиентов. |
accumulator_ref | Указатель на условный сумматор, созданный подклассами. |
| Атрибуты | |
|---|---|
accumulator_ref | Ссылка на базовый сумматор. |
dtype | Тип данных градиентов, накапливаемых этим сумматором. |
name | Имя базового сумматора. |
Методы
num_accumulated
num_accumulated(
name=None
)
Количество градиентов, которые в настоящее время были агрегированы в сумматоре.
| Аргументы | |
|---|---|
name | Необязательное имя операции. |
| Возвращаемые значения | |
|---|---|
| Количество накопленных градиентов в данный момент в сумматоре. |
set_global_step
set_global_step(
new_global_step, name=None
)
Устанавливает глобальный шаг времени сумматора.
Операция записывает предупреждение, если мы пытаемся установить шаг времени меньше, чем текущий шаг сумматора.
| Аргументы | |
|---|---|
new_global_step | Значение нового шага времени. Может быть переменной или константой. |
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/r1.15/api_docs/python/tf/ConditionalAccumulatorBase