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.
Включает переменные слотов и дополнительные глобальные переменные, созданные оптимизатором в текущей стандартной графе.
| Возвращаемое значение | |
|---|---|
| Список переменных. |
Переменные класса
-
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