tf.contrib.opt.VariableClippingOptimizer
Оборачивающий оптимизатор, который обрезает норму указанных переменных после обновления.
Наследуется от: Optimizer
tf.contrib.opt.VariableClippingOptimizer(
opt, vars_to_clip_dims, max_norm, use_locking=False,
colocate_clip_ops_with_vars=False, name='VariableClipping'
)
Этот оптимизатор делегирует все аспекты вычисления и применения градиента базовому оптимизатору. После применения градиентов этот оптимизатор обрезает переменную, чтобы максимальная норма L2 по указанным измерениям была заданной.
Обратите внимание: это существенно отличается от обрезания нормы градиентов.
Несколько экземпляров VariableClippingOptimizer могут быть объединены для задания различных максимальных норм для различных подмножеств переменных.
Это более эффективно во время обслуживания, чем использование нормализации во время поиска по вложениям, но при этом обучение становится более затратным и гарантированные нормы меньше.
| Аргументы | |
|---|---|
opt | Фактический оптимизатор, который будет использоваться для вычисления и применения градиентов. Должен быть одним из классов Optimizer. |
vars_to_clip_dims | Словарь с переменными в качестве ключей и списками измерений для вычисления нормы L2 в качестве значений. См. tf.clip_by_norm для получения более подробной информации. |
max_norm | Норма L2, до которой следует обрезать все указанные переменные. |
use_locking | Если True использовать блокировки для операций обновления обрезки. |
colocate_clip_ops_with_vars | Если True, попытаться разместить операции clip norm вместе с соответствующей переменной. |
name | Необязательный префикс имени для операций, созданных при применении градиентов. По умолчанию "VariableClipping". |
Методы
apply_gradients
apply_gradients(
grads_and_vars, global_step=None, name=None
)
Применить градиенты к переменным.
Это вторая часть minimize(). Она возвращает Operation, который применяет градиенты.
| Аргументы | |
|---|---|
grads_and_vars | Список пар (градиент, переменная), как возвращается compute_gradients(). |
global_step | Необязательный Variable для инкремента на единицу после обновления переменных. |
name | Необязательное имя возвращаемой операции. По умолчанию имя, переданное конструктору Optimizer . |
| Возвращаемое значение | |
|---|---|
Операция, которая применяет указанные градиенты. Если global_step не равно None, эта операция также увеличивает global_step. |
| Исключения | |
|---|---|
TypeError | Если grads_and_vars имеет неправильный формат. |
ValueError | Если ни у одной из переменных нет градиентов. |
RuntimeError | Если следует использовать _distributed_apply() вместо этого. |
compute_gradients
compute_gradients(
*args, **kwargs
)
Вычислить градиенты 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. |
colocate_gradients_with_ops | Если True, попытаться разместить градиенты вместе с соответствующей операцией. |
grad_loss | Необязательно. Объект Tensor, содержащий градиент, вычисленный для loss. |
| Возвращаемое значение | |
|---|---|
Список пар (градиент, переменная). Переменная всегда присутствует, но градиент может быть None. |
| Исключения | |
|---|---|
TypeError | Если var_list содержит что-либо кроме Variable объектов. |
ValueError | Если некоторые аргументы некорректны. |
RuntimeError | Если вызов осуществляется с включенным режимом eager выполнения, а loss не является вызываемым объектом. |
Совместимость с режимом eager
Если режим eager выполнения включен, gate_gradients, aggregation_method, и colocate_gradients_with_ops игнорируются.
get_name
get_name()
get_slot
get_slot(
*args, **kwargs
)
Возвращает слот с именем name, созданный для var оптимизатором.
Некоторые подклассы Optimizer используют дополнительные переменные. Например, Momentum и Adagrad используют переменные для накопления обновлений. Этот метод предоставляет доступ к этим Variable объектам, если они вам по какой-либо причине нужны.
Используйте get_slot_names() для получения списка имён слотов, созданных Optimizer.
| Аргументы | |
|---|---|
var | Переменная, переданная в minimize() или apply_gradients() . |
name | Строка. |
| Возвращаемое значение | |
|---|---|
Variable для слота, если он был создан, None в противном случае. |
get_slot_names
get_slot_names(
*args, **kwargs
)
Возвращает список имён слотов, созданных оптимизатором.
См. 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/VariableClippingOptimizer