tf.contrib.opt.RegAdagradOptimizer
RegAdagrad: Adagrad с обновлениями, которые необязательно пропускают обновление слотов.
Наследуется от: AdagradOptimizer
tf.contrib.opt.RegAdagradOptimizer(
learning_rate, initial_accumulator_value=0.1, use_locking=False,
name='RegAdagrad'
)
Это предназначено для решения проблемы дополнительных регуляризационных членов в функции потерь, влияющих на распад скорости обучения и вызывающих переплетение гиперпараметров. Пример использования:
loss = tf.nn.cross_entropy(x, labels) reg_loss = reg_strength * tf.reduce_sum(x * x) opt = tf.contrib.opt.RegAdagradOptimizer(learning_rate) loss_update = opt.minimize(loss) with opt.avoid_updating_slots(): reg_update = opt.minimize(reg_loss) total_update = tf.group([loss_update, reg_update])
# ...
sess.run(total_update, ...)
| Аргументы | |
|---|---|
learning_rate | Значение типа `Tensor` или число с плавающей точкой. Скорость обучения. |
initial_accumulator_value | Число с плавающей точкой. Начальное значение для аккумуляторов, должно быть положительным. |
use_locking | Если True использовать блокировки для операций обновления. |
name | Необязательный префикс имени для операций, созданных при применении градиентов. По умолчанию "Adagrad". |
| Исключения | |
|---|---|
ValueError | Если initial_accumulator_value некорректен. |
Методы
apply_gradients
apply_gradients(
grads_and_vars, global_step=None, name=None
)
Применить градиенты к переменным.
Это вторая часть minimize(). Она возвращает операцию, которая применяет градиенты.
| Аргументы | |
|---|---|
grads_and_vars | Список пар (градиент, переменная), возвращённый compute_gradients(). |
global_step | Необязательное значение, которое увеличивается на единицу после обновления переменных. |
name | Необязательное имя возвращаемой операции. По умолчанию совпадает с именем, переданным конструктору Optimizer. |
| Возвращает | |
|---|---|
Операция, применяющая указанные градиенты. Если global_step не равно None, эта операция также увеличивает global_step. |
| Исключения | |
|---|---|
TypeError | Если grads_and_vars имеет неправильный формат. |
ValueError | Если у ни одной из переменных нет градиента. |
RuntimeError | Если следует использовать _distributed_apply() вместо этого. |
avoid_updating_slots
@tf_contextlib.contextmanager avoid_updating_slots()
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`, `Variable` или `None`, если для данной переменной нет градиента.
| Аргументы | |
|---|---|
loss | `Tensor`, содержащий значение для минимизации, или вызываемая функция без аргументов, возвращающая значение для минимизации. При включённом режиме выполнения eager, это должна быть функция. |
var_list | Необязательный список или кортеж переменных `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(
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 | Необязательное значение, которое увеличивается на единицу после обновления переменных. |
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`. |
Совместимость с режимом eager выполнения
При включенном режиме немедленного выполнения 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/RegAdagradOptimizer