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/versions/r2.9/api_docs/python/tf/compat/v1/ConditionalAccumulatorBase