Spec-Zone.ru › TensorFlow 2.9

tf.compat.v1.train.ProximalAdagradOptimizer

Оптимизатор, реализующий алгоритм Proximal Adagrad.

Наследуется от: Optimizer

tf.compat.v1.train.ProximalAdagradOptimizer(
    learning_rate,
    initial_accumulator_value=0.1,
    l1_regularization_strength=0.0,
    l2_regularization_strength=0.0,
    use_locking=False,
    name='ProximalAdagrad'
)

Ссылки:

Adaptive Subgradient Methods for Online Learning and Stochastic Optimization: Duchi et al., 2011 (pdf) Efficient Learning using Forward-Backward Splitting: Duchi et al., 2009 (pdf)

Аргументы
learning_rate Значение типа `Tensor` или число с плавающей точкой. Скорость обучения.
initial_accumulator_value Число с плавающей точкой. Начальное значение для аккумуляторов, должно быть положительным.
l1_regularization_strength Число с плавающей точкой, должно быть больше или равно нулю.
l2_regularization_strength Число с плавающей точкой, должно быть больше или равно нулю.
use_locking Если True, использовать блокировки для операций обновления.
name Необязательный префикс имени для операций, созданных при применении градиентов. По умолчанию "Adagrad".
Исключения
ValueError Если initial_accumulator_value некорректное.

Методы

apply_gradients

Посмотреть исходный код

apply_gradients(
    grads_and_vars, global_step=None, name=None
)

Применить градиенты к переменным.

Это вторая часть minimize(). Она возвращает операцию, применяющую градиенты.

@compatibility(TF2)

Как сопоставить аргументы

Имя аргумента TF1 Имя аргумента TF2 Примечание
grads_and_vars grads_and_vars -
global_step Не поддерживается. Используйте optimizer.iterations
name name. -
Аргументы
grads_and_vars Список пар (градиент, переменная), как возвращается compute_gradients().
global_step Необязательный `Tensor`, увеличивающийся на единицу после обновления переменных.
name Необязательное имя возвращаемой операции. По умолчанию имя, переданное в конструктор Optimizer.
Возвращаемое значение
Операция, применяющая указанные градиенты. Если `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

Внимание: Этот API был разработан для TensorFlow v1. Продолжайте чтение для получения подробной информации о том, как мигрировать с этого API в эквивалент TensorFlow v2. См. руководство по миграции TensorFlow v1 в TensorFlow v2 для инструкций по миграции остальной части вашего кода.

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, должен быть вызываемой функцией. 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`, содержащий вычисленный градиент для Tensor.

Возвращаемое значение: Список пар (градиент, переменная). Переменная всегда присутствует, но градиент может быть None.

Исключения: TypeError: Если var_list содержит что-либо кроме объектов Variable. ValueError: Если некоторые аргументы некорректны. RuntimeError: Если вызвана при включенном режиме eager исполнения, и loss не является вызываемой функцией.

@compatibility(eager) При включенном режиме eager исполнения gate_gradients, aggregation_method, и colocate_gradients_with_ops игнорируются.

Описание

Это первая часть minimize(). Она возвращает список пар (градиент, переменная), где "градиент" — это градиент для "переменной". Обратите внимание, что "градиент" может быть `Tensor`, `IndexedSlices` или `None`, если для данной переменной нет градиента.

get_name

Посмотреть исходный код

get_name()

get_slot

Посмотреть исходный код

get_slot(
    var, name
)

Возвращает слот с именем name, созданный для var оптимизатором.

Некоторые подклассы оптимизаторов используют дополнительные переменные. Например, Momentum и Adagrad используют переменные для накопления обновлений. Этот метод предоставляет доступ к этим переменным, если они вам нужны по какой-либо причине.

Используйте get_slot_names() для получения списка имен слотов, созданных оптимизатором Optimizer.

Аргументы
var Переменная, переданная в minimize() или apply_gradients().
name Строка.
Возвращаемое значение
Переменная слота, если она была создана, `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 Необязательный `Tensor` для увеличения на единицу после обновления переменных.
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/versions/r2.9/api_docs/python/tf/compat/v1/train/ProximalAdagradOptimizer

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API