noam_decay_cn.rst 1.2 KB
Newer Older
H
Hao Wang 已提交
1 2 3 4 5 6 7
.. _cn_api_fluid_layers_noam_decay:

noam_decay
-------------------------------

.. py:function:: paddle.fluid.layers.noam_decay(d_model,warmup_steps)

X
xiaoting 已提交
8 9 10
Noam衰减方法

noam衰减的numpy实现如下:
H
Hao Wang 已提交
11 12 13

.. code-block:: python

X
xiaoting 已提交
14
    import paddle.fluid as fluid
H
Hao Wang 已提交
15 16 17 18 19 20 21 22 23 24 25 26 27
    import numpy as np
    # 设置超参数
    d_model = 2
    current_steps = 20
    warmup_steps = 200
    # 计算
    lr_value = np.power(d_model, -0.5) * np.min([
                           np.power(current_steps, -0.5),
                           np.power(warmup_steps, -1.5) * current_steps])

请参照 `attention is all you need <https://arxiv.org/pdf/1706.03762.pdf>`_

参数:
X
xiaoting 已提交
28 29
    - **d_model** (Variable|int) - 模型的输入、输出向量特征维度。类型可设置为标量Tensor,或int值。
    - **warmup_steps** (Variable|int) - 预热步数,类型可设置为标量Tensor,或int值。
H
Hao Wang 已提交
30 31 32

返回:衰减的学习率

X
xiaoting 已提交
33 34
返回类型: Variable

H
Hao Wang 已提交
35 36 37 38
**代码示例**:

.. code-block:: python

X
xiaoting 已提交
39
        import paddle.fluid as fluid
H
Hao Wang 已提交
40 41 42 43 44 45 46 47 48 49 50
        warmup_steps = 100
        learning_rate = 0.01
        lr = fluid.layers.learning_rate_scheduler.noam_decay(
                       1/(warmup_steps *(learning_rate ** 2)),
                       warmup_steps)