Spec-Zone.ru › TensorFlow 1.15

tf.contrib.opt.LazyAdamGSOptimizer

Вариант оптимизатора Adam, который более эффективно обрабатывает разреженные обновления.

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

tf.contrib.opt.LazyAdamGSOptimizer(
    global_step=0, learning_rate=0.001, beta1=0.9, beta2=0.999, epsilon=1e-08,
    use_locking=False, name='Adam'
)

Ветвь от tf.contrib.opt.LazyAdamGSOptimizer. Единственное отличие состоит в том, что для вычисления накопителей beta1 и beta2 передаётся глобальный шаг, вместо того, чтобы оптимизатор хранил свои собственные независимые накопители beta1 и beta2 как переменные, не являющиеся слотами.

Оригинальный алгоритм Adam поддерживает два накопителя скользящего среднего для каждой обучаемой переменной; накопители обновляются на каждом шаге. Этот класс предоставляет более ленивую обработку обновлений градиента для разреженных переменных. Он обновляет накопители скользящего среднего только для индексов разреженных переменных, которые появляются в текущей партии, а не для всех индексов. По сравнению с оригинальным оптимизатором Adam, он может существенно улучшить производительность обучения модели для некоторых приложений. Однако он обеспечивает немного другие семантику, чем оригинальный алгоритм Adam, и может привести к другим эмпирическим результатам.

Аргументы
global_step Переменная TensorFlow, указывающая шаг.
learning_rate Тензор или значение с плавающей точкой. Скорость обучения.
beta1 Значение с плавающей точкой или постоянный тензор с плавающей точкой. Скорость экспоненциального затухания для оценок первого момента.
beta2 Значение с плавающей точкой или постоянный тензор с плавающей точкой. Скорость экспоненциального затухания для оценок второго момента.
epsilon Небольшая константа для обеспечения числовой устойчивости. Эта величина epsilon — «epsilon hat» в статье Kingma и Ba (в формуле перед разделом 2.1), а не epsilon в алгоритме 1 статьи.
use_locking Если True, использовать блокировки для операций обновления.
name Необязательное имя для операций, созданных при применении градиентов. По умолчанию «Adam». @compatibility(eager) Когда включено выполнение eager, learning_rate, beta1, beta2, и epsilon могут быть функциями, которые принимают на вход ничего и возвращают фактическое значение для использования. Это может быть полезно для изменения этих значений в разных вызовах функций оптимизатора. @end_compatibility

Методы

apply_gradients

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

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

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

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

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

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

Аргументы
loss Тензор, содержащий значение для минимизации, или вызываемая функция, принимающая на вход ничего и возвращающая значение для минимизации. При включенном 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 не является вызываемой функцией.

Совместимость с Eager

Когда включено eager execution, gate_gradients, aggregation_method, и colocate_gradients_with_ops игнорируются.

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

© 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/r1.15/api_docs/python/tf/contrib/opt/LazyAdamGSOptimizer

Spec-Zone.ru

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