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