Spec-Zone.ru › TensorFlow 1.15

tf.contrib.opt.GGTOptimizer

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

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

tf.contrib.opt.GGTOptimizer(
    learning_rate=0.001, beta1=0.9, use_locking=False, name='GGT', window=10,
    eps=0.0001, svd_eps=1e-06, sigma_eps=0.01
)

GGT имеет преимущество перед sgd и adam на больших моделях с плохой обусловленностью, например, языковых моделях и CNN, см. [ABCHSZZ 2018].

Аргументы
learning_rate Вещественный гиперпараметр. Скорость обучения.
beta1 Вещественный гиперпараметр. Скорость экспоненциального затухания для оценок первого момента.
use_locking Если True, использовать блокировки для операций обновления.
name Необязательное имя для операций, создаваемых при применении градиентов. По умолчанию "GGT".
window Целочисленный гиперпараметр. Количество первых моментов для вычисления адаптивного предобусловителя.
eps Вещественный гиперпараметр. Используется для обрезания малых собственных значений матрицы ковариации градиента.
svd_eps Вещественный гиперпараметр. Используется для стабилизации SVD.
sigma_eps Вещественный гиперпараметр. Используется для регуляризации обращения матрицы.

Методы

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 Если ни одна из переменных не имеет градиента.

compute_gradients

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

compute_gradients(
    loss, var_list=None, gate_gradients=GATE_OP, aggregation_method=None,
    grad_loss=None, stop_gradients=None, scale_loss_by_num_replicas=False
)

Вычисление градиентов loss для переменных в var_list.

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

Аргументы
loss Тензор, содержащий значение для минимизации, или вызываемая функция без аргументов, возвращающая значение для минимизации. При включённом режиме выполнения eager, это должна быть функция.
var_list Необязательный список или кортеж tf.Variable для обновления, чтобы минимизировать loss. По умолчанию - список переменных, собранных в графе по ключу GraphKeys.TRAINABLE_VARIABLES.
gate_gradients Способ управления вычислением градиентов. Может быть GATE_NONE, GATE_OP, или GATE_GRAPH.
aggregation_method Указывает метод комбинирования градиентных слагаемых. Действительные значения определены в классе AggregationMethod.
grad_loss Необязательно. Tensor содержащий вычисленный градиент для loss.
stop_gradients Необязательно. Тензор или список тензоров, через которые не нужно дифференцировать.
scale_loss_by_num_replicas Необязательный булевый параметр. Если True, масштабирует функцию потерь в соответствии с числом реплик. УСТАРЕВШЕЕ и обычно больше не требуется.
Возвращаемое значение
Список пар (градиент, переменная). Переменная всегда присутствует, но градиент может быть None.
Исключения
TypeError Если var_list содержит что-либо кроме Variable объектов.
ValueError Если некоторые аргументы неверны.
RuntimeError Если вызов осуществляется с включённым eager выполнением и loss не является вызываемой функцией.

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

При включенном eager выполнении gate_gradients, и aggregation_method игнорируются.

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, name=None, grad_loss=None, stop_gradients=None,
    scale_loss_by_num_replicas=False
)

Добавить операции для минимизации 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.
name Необязательное имя возвращаемой операции.
grad_loss Необязательно. Tensor содержащий вычисленный градиент для loss.
stop_gradients Необязательно. Тензор или список тензоров, через которые не нужно дифференцировать.
scale_loss_by_num_replicas Необязательный булевый параметр. Если True, масштабирует функцию потерь в соответствии с числом реплик. УСТАРЕВШЕЕ и обычно больше не требуется.
Возвращаемое значение
Операция, обновляющая переменные в var_list. Если global_step не равно None, эта операция также увеличивает global_step.
Исключения
ValueError Если некоторые переменные не являются Variable объектами.

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

При включенном режиме немедленного выполнения loss должно быть функцией Python, которая принимает элементы var_list в качестве аргументов и вычисляет значение, которое необходимо минимизировать. Если var_list равно None, то loss не должна принимать никаких аргументов. Минимизация (и вычисление градиента) выполняется относительно элементов var_list если они не равны None, иначе относительно любых обучаемых переменных, созданных во время выполнения функции loss. gate_gradients, aggregation_method, и grad_loss игнорируются при включенном режиме немедленного выполнения.

variables

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

variables()

Список переменных, которые кодируют текущее состояние Optimizer.

Включает переменные слотов и дополнительные глобальные переменные, созданные оптимизатором в текущей стандартной графе.

class="responsive fixed orange">
Возвращаемое значение
Список переменных.

Переменные класса

  • 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/GGTOptimizer

Spec-Zone.ru

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