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 включён, эта функция возвращает функцию, которая, в свою очередь, возвращает тензор затухающего темпа обучения. Это может быть полезно для изменения значения темпа обучения при разных вызовах функций оптимизатора.
© 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