mindspore.nn.WarmUpLR ====================== .. py:class:: mindspore.nn.WarmUpLR(learning_rate, warmup_steps) 预热学习率。 对于当前step,计算学习率的公式为: .. math:: warmup\_learning\_rate = learning\_rate * tmp\_step / warmup\_steps 其中, .. math:: tmp\_step=min(current\_step, warmup\_steps) 参数: - **learning_rate** (float) - 学习率的初始值。 - **warmup_steps** (int) - 学习率warmup的step数。 输入: - **global_step** (Tensor) - 当前step数,即current_step。 输出: 标量Tensor。当前step的学习率值。 异常: - **TypeError** - `learning_rate` 不是float。 - **TypeError** - `warmup_steps` 不是int。 - **ValueError** - `warmup_steps` 小于1。 - **ValueError** - `learning_rate` 小于或等于0。