tf.compat.v1.train.AdagradDAOptimizer
Алгоритм Adagrad Dual Averaging для разреженных линейных моделей.
Наследуется от: Optimizer
tf.compat.v1.train.AdagradDAOptimizer(
learning_rate,
global_step,
initial_gradient_squared_accumulator_value=0.1,
l1_regularization_strength=0.0,
l2_regularization_strength=0.0,
use_locking=False,
name='AdagradDA'
)
Этот оптимизатор заботится о регуляризации несхватаемых признаков в мини-пачке, обновляя их, когда они встречаются, с помощью явной формулы обновления, эквивалентной обновлению на каждой мини-пачке.
AdagradDA обычно используется, когда требуется большая разреженность в обученной модели. Этот оптимизатор гарантирует разреженность только для линейных моделей. Будьте осторожны, используя AdagradDA для глубоких сетей, так как потребуется тщательная инициализация аккумуляторов градиента для обучения.
| Ссылки | |
|---|---|
| Методы адаптивных субградиентных методов для онлайн-обучения и стохастической оптимизации: Duchi et al., 2011 (pdf) |
| Аргументы | |
|---|---|
learning_rate | Числовой Tensor или значение с плавающей точкой. Скорость обучения. |
global_step | Tensor, содержащий текущий номер шага обучения. |
initial_gradient_squared_accumulator_value | Числовое значение с плавающей точкой. Начальное значение для аккумуляторов, должно быть положительным. |
l1_regularization_strength | Значение с плавающей точкой, должно быть не меньше нуля. |
l2_regularization_strength | Значение с плавающей точкой, должно быть не меньше нуля. |
use_locking | Если True использовать блокировки для операций обновления. |
name | Необязательное префиксное имя для операций, созданных при применении градиентов. По умолчанию "AdagradDA". |
| Исключения | |
|---|---|
ValueError | Если initial_gradient_squared_accumulator_value неверно. |
Методы
apply_gradients
apply_gradients(
grads_and_vars,
global_step=None,
name=None,
skip_gradients_aggregation=False
)
Применение градиентов к переменным.
Это вторая часть minimize(). Возвращает Operation, применяющий градиенты.
@compatibility(TF2)
Как сопоставить аргументы
| Имя аргумента TF1 | Имя аргумента TF2 | Примечание |
|---|---|---|
grads_and_vars | grads_and_vars | - |
global_step | Не поддерживается. | Используйте optimizer.iterations
|
name | name. | - |
| Аргументы | |
|---|---|
grads_and_vars | Список пар (градиент, переменная), возвращаемых compute_gradients(). |
global_step | Необязательное Variable, увеличивающееся на единицу после обновления переменных. |
name | Необязательное имя возвращаемой операции. По умолчанию имя, переданное в конструктор Optimizer. |
skip_gradients_aggregation | Если True, агрегация градиентов не выполняется внутри оптимизатора. Обычно этот аргумент устанавливается в True, когда вы пишете пользовательский код агрегирования градиентов вне оптимизатора. |
| Возвращаемое значение | |
|---|---|
Operation, применяющий указанные градиенты. Если global_step не равно None, эта операция также увеличивает global_step. |
| Исключения | |
|---|---|
TypeError | Если grads_and_vars имеет неправильный формат. |
ValueError | Если ни у одной из переменных нет градиента. |
RuntimeError | Если следует использовать _distributed_apply(). |
compute_gradients
compute_gradients(
loss,
var_list=None,
gate_gradients=GATE_OP,
aggregation_method=None,
colocate_gradients_with_ops=False,
grad_loss=None
)
Вычисление градиентов loss для переменных в var_list.
Переход на TF2
tf.keras.optimizers.Optimizer в TF2 не предоставляет метод compute_gradients, и вы должны использовать tf.GradientTape для получения градиентов:
@tf.function
def train step(inputs):
batch_data, labels = inputs
with tf.GradientTape() as tape:
predictions = model(batch_data, training=True)
loss = tf.keras.losses.CategoricalCrossentropy(
reduction=tf.keras.losses.Reduction.NONE)(labels, predictions)
gradients = tape.gradient(loss, model.trainable_variables)
optimizer.apply_gradients(zip(gradients, model.trainable_variables))
Аргументы: loss: Tensor, содержащий значение для минимизации или вызываемый объект без аргументов, возвращающий значение для минимизации. При включённом режиме Eager Execution он должен быть вызываемым. var_list: Необязательный список или кортеж tf.Variable для обновления, чтобы минимизировать loss. По умолчанию список переменных, собранных в графе под ключом GraphKeys.TRAINABLE_VARIABLES. gate_gradients: Способ управления вычислением градиентов. Может быть GATE_NONE, GATE_OP или GATE_GRAPH. aggregation_method: Указывает метод комбинирования градиентных членов. Допустимые значения определены в классе AggregationMethod. colocate_gradients_with_ops: Если True, попытаться разместить градиенты вместе с соответствующей операцией. grad_loss: Необязательно. Tensor, содержащий градиент, вычисленный для loss.
Возвращаемое значение: Список пар (градиент, переменная). Переменная всегда присутствует, но градиент может быть None.
Исключения: TypeError: Если var_list содержит что-либо кроме Variable объектов. ValueError: Если некоторые аргументы неверны. RuntimeError: Если вызвано при включенном режиме Eager Execution и loss не является вызываемым.
@compatibility(eager) При включённом режиме Eager Execution, gate_gradients, aggregation_method и colocate_gradients_with_ops игнорируются.
Описание
Это первая часть minimize(). Возвращает список пар (градиент, переменная), где «градиент» — градиент для «переменной». Обратите внимание, что «градиент» может быть Tensor, IndexedSlices или None, если для данной переменной нет градиента.
get_name
get_name()
get_slot
get_slot(
var, name
)
Возвращает слот с именем name, созданный для var оптимизатором.
Некоторые подклассы Optimizer используют дополнительные переменные. Например, Momentum и Adagrad используют переменные для накопления обновлений. Этот метод предоставляет доступ к этим Variable объектам, если вам они понадобятся по какой-либо причине.
Используйте get_slot_names(), чтобы получить список имён слотов, созданных Optimizer.
| Аргументы | |
|---|---|
var | Переменная, переданная в minimize() или apply_gradients(). |
name | Строка. |
| Возвращаемое значение | |
|---|---|
Variable для слота, если он был создан, None в противном случае. |
get_slot_names
get_slot_names()
Возвращает список имён слотов, созданных Optimizer.
См. get_slot().
| Возвращаемое значение | |
|---|---|
| Список строк. |
minimize
minimize(
loss,
global_step=None,
var_list=None,
gate_gradients=GATE_OP,
aggregation_method=None,
colocate_gradients_with_ops=False,
name=None,
grad_loss=None
)
Добавляет операции для минимизации loss, обновляя var_list.
Этот метод просто объединяет вызовы compute_gradients() и apply_gradients(). Если вы хотите обработать градиент перед его применением, вызовите compute_gradients() и apply_gradients() явно вместо использования этой функции.
| Аргументы | |
|---|---|
loss | Tensor, содержащий значение для минимизации. |
global_step | Необязательный Variable, который увеличивается на единицу после обновления переменных. |
var_list | Необязательный список или кортеж объектов Variable для обновления, чтобы минимизировать loss. По умолчанию используется список переменных, собранных в графе под ключом GraphKeys.TRAINABLE_VARIABLES. |
gate_gradients | Способ управления вычислением градиентов. Может быть GATE_NONE, GATE_OP или GATE_GRAPH. |
aggregation_method | Указывает метод объединения членов градиента. Допустимые значения определены в классе AggregationMethod. |
colocate_gradients_with_ops | Если True, попытаться разместить градиенты вместе с соответствующей операцией. |
name | Необязательное имя возвращаемой операции. |
grad_loss | Необязательно. Tensor, содержащий градиент, вычисленный для loss. |
| Возвращаемое значение | |
|---|---|
Операция, обновляющая переменные в var_list. Если global_step не было None, эта операция также увеличивает global_step. |
| Исключения | |
|---|---|
ValueError | Если некоторые переменные не являются объектами Variable. |
Совместимость с немедленным выполнением
Когда включено немедленное выполнение, loss должно быть функцией Python, которая не принимает аргументов и вычисляет значение, которое нужно минимизировать. Минимизация (и вычисление градиента) выполняется относительно элементов var_list, если оно не равно None, в противном случае относительно любых обучаемых переменных, созданных во время выполнения функции loss. gate_gradients, aggregation_method, colocate_gradients_with_ops и grad_loss игнорируются при включенном немедленном выполнении.
variables
variables()
Список переменных, которые кодируют текущее состояние Optimizer.
Включает переменные слотов и дополнительные глобальные переменные, созданные оптимизатором в текущем графике по умолчанию.
| Возвращаемое значение | |
|---|---|
| Список переменных. |
| Переменные класса | |
|---|---|
| GATE_GRAPH | 2 |
| GATE_NONE | 0 |
| GATE_OP | 1 |
© 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/train/AdagradDAOptimizer