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