Python tf.compat.v1.train.linear_cosine_decay用法及代码示例

对学习率应用线性余弦衰减。

用法

tf.compat.v1.train.linear_cosine_decay(
    learning_rate, global_step, decay_steps, num_periods=0.5, alpha=0.0, beta=0.001,
    name=None
)

请注意，线性余弦衰减比余弦衰减更具侵略性，通常可以使用更大的初始学习率。

在训练模型时，通常建议随着训练的进行降低学习率。此函数将线性余弦衰减函数应用于提供的初始学习率。它需要一个global_step 值来计算衰减的学习率。您可以只传递一个 TensorFlow 变量，该变量在每个训练步骤中递增。

该函数返回衰减的学习率。它被计算为：

global_step = min(global_step, decay_steps)
linear_decay = (decay_steps - global_step) / decay_steps)
cosine_decay = 0.5 * (
    1 + cos(pi * 2 * num_periods * global_step / decay_steps))
decayed = (alpha + linear_decay) * cosine_decay + beta
decayed_learning_rate = learning_rate * decayed

示例用法：

decay_steps = 1000
lr_decayed = linear_cosine_decay(learning_rate, global_step, decay_steps)

参考：

带有强化学习的神经优化器搜索：Bello 等人，2017 (pdf) 带有热重启的随机梯度下降：Loshchilov 等人，2017 (pdf)

eager模式兼容性

当启用即刻执行时，此函数返回一个函数，该函数又返回衰减的学习率张量。这对于在优化器函数的不同调用中更改学习率值很有用。