LRScheduler
学习率策略的基类。定义了所有学习率调整策略的公共接口。
目前在paddle中基于该基类,已经实现了12种策略,分别为:
NoamDecay
: 诺姆衰减,相关算法请参考 《Attention Is All You Need》 。请参考ExponentialDecay
: 指数衰减,即每次将当前学习率乘以给定的衰减率得到下一个学习率。请参考 ExponentialDecayNaturalExpDecay
: 自然指数衰减,即每次将当前学习率乘以给定的衰减率的自然指数得到下一个学习率。请参考InverseTimeDecay
: 逆时间衰减,即得到的学习率与当前衰减次数成反比。请参考 InverseTimeDecayPolynomialDecay
: 多项式衰减,即得到的学习率为初始学习率和给定最终学习之间由多项式计算权重定比分点的插值。请参考 cn_api_paddle_optimizer_lr_PolynomialDecayPiecewiseDecay
: 分段衰减,即由给定step数分段呈阶梯状衰减,每段内学习率相同。请参考CosineAnnealingDecay
: 余弦式衰减,即学习率随step数变化呈余弦函数周期变化。请参考 CosineAnnealingDecayLinearWarmup
: 学习率随step数线性增加到指定学习率。请参考MultiStepDecay
: 学习率在特定的step数时进行衰减,需要指定衰减时的节点位置。请参考 MultiStepDecayReduceOnPlateau
: 学习率根据当前监控指标(一般为loss)来进行自适应调整,当loss趋于稳定时衰减学习率。请参考
你可以继承该基类实现任意的学习率策略,导出基类的方法为 form paddle.optimizer.lr import LRScheduler
, 必须要重写该基类的 get_lr()
函数,否则会抛出 NotImplementedError
异常。
参数:
learning_rate (float, 可选) - 初始学习率,数据类型为Python float。
last_epoch (int,可选) - 上一轮的轮数,重启训练时设置为上一轮的epoch数。默认值为 -1,则为初始学习率。
verbose (bool,可选) - 如果是
True
,则在每一轮更新时在标准输出 stdout 输出一条信息。默认值为 。
返回:用于调整学习率的实例对象。
代码示例
step
( epoch=None )
step函数需要在优化器的 optimizer.step() 函数之后调用,调用之后将会根据epoch数来更新学习率,更新之后的学习率将会在优化器下一轮更新参数时使用。
参数:
- epoch (int,可选)- 指定具体的epoch数。默认值None,此时将会从-1自动累加
epoch
数。
返回:无。
代码示例 :
请参考 基类LRScheduler
的任意子类实现,这里以 StepLR
为例进行了示例:
import paddle
import numpy as np
x = np.random.uniform(-1, 1, [10, 10]).astype("float32")
linear = paddle.nn.Linear(10, 10)
sgd = paddle.optimizer.SGD(learning_rate=scheduler, parameters=linear.parameters())
for epoch in range(20):
for batch_id in range(2):
x = paddle.to_tensor(x)
loss = paddle.mean(out)
loss.backward()
sgd.step()
sgd.clear_gradients()
scheduler.step() # If you update learning rate each step
get_lr
( )
如果一个子类继承了 基类LRScheduler
,则用户必须重写方法 get_lr()
,否则,将会抛出 NotImplementedError
异常,
上述给出了实现 StepLR
的一个简单示例。
_state_keys
( )
如果需要改变默认的行为,用户需要重写该方法,来重新定义字典 self.keys
,一般无需重新设置。