Spec-Zone.ru › TensorFlow 2.4

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 Execution

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

© 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/r2.4/api_docs/python/tf/compat/v1/train/inverse_time_decay

Spec-Zone.ru

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