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