tf.compat.v1.ConditionalAccumulatorBase
Условный аккумулятор для агрегирования градиентов.
tf.compat.v1.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 | Необязательное имя операции. |
| Возвращает | |
|---|---|
| Операция, устанавливающая шаг времени аккумулятора. |
© 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/compat/v1/ConditionalAccumulatorBase