Spec-Zone.ru › TensorFlow

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

Spec-Zone.ru

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