Spec-Zone.ru › TensorFlow 2.9

tf.compat.v1.train.inverse_time_decay

Применяет обратное убывание во времени к начальной скорости обучения.

tf.compat.v1.train.inverse_time_decay(
    learning_rate,
    global_step,
    decay_steps,
    decay_rate,
    staircase=False,
    name=None
)

При обучении модели часто рекомендуется уменьшать скорость обучения по мере прогресса обучения. Эта функция применяет обратную функцию убывания к предоставленной начальной скорости обучения. Она требует global_step значения для вычисления уменьшенной скорости обучения. Вы можете просто передать переменную TensorFlow, которую вы инкрементируете на каждом шаге обучения.

Функция возвращает уменьшенную скорость обучения. Она вычисляется как:

decayed_learning_rate = learning_rate / (1 + decay_rate * global_step /
decay_step)

или, если staircase равно True, как:

decayed_learning_rate = learning_rate / (1 + decay_rate * floor(global_step /
decay_step))

Пример: убывание 1/t со скоростью 0,5:

...
global_step = tf.Variable(0, trainable=False)
learning_rate = 0.1
decay_steps = 1.0
decay_rate = 0.5
learning_rate = tf.compat.v1.train.inverse_time_decay(learning_rate,
global_step,
decay_steps, decay_rate)

# Passing global_step to minimize() will increment it at each step.
learning_step = (
    tf.compat.v1.train.GradientDescentOptimizer(learning_rate)
    .minimize(...my loss..., global_step=global_step)
)
Аргументы
learning_rate Скалярное float32 или float64 Tensor или число Python. Начальная скорость обучения.
global_step Число Python. Глобальный шаг для вычисления убывания. Должно быть неотрицательным.
decay_steps Частота применения убывания.
decay_rate Число Python. Скорость убывания.
staircase Применять убывание дискретными ступенями, а не непрерывно.
name Строка. Необязательное имя операции. По умолчанию 'InverseTimeDecay'.
Возвращает
Скаляр Tensor того же типа, что и learning_rate. Уменьшенная скорость обучения.
Исключения
ValueError если global_step не предоставлено.

совместимость с eager выполнением

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

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/compat/v1/train/inverse_time_decay

Spec-Zone.ru

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