Skip to content

Commit a84bd14

Browse files
authored
在KaimingNormal/Uniform中加入mode参数 (#7324)
1 parent 9f635ff commit a84bd14

File tree

4 files changed

+10
-6
lines changed

4 files changed

+10
-6
lines changed

docs/api/paddle/nn/initializer/KaimingNormal_cn.rst

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
KaimingNormal
44
-------------------------------
55

6-
.. py:class:: paddle.nn.initializer.KaimingNormal(fan_in=None, negative_slope=0.0, nonlinearity='relu')
6+
.. py:class:: paddle.nn.initializer.KaimingNormal(fan_in=None, negative_slope=0.0, nonlinearity='relu', mode='fan_in')
77
88
99
@@ -22,6 +22,7 @@ Kaiming 正态分布方式的权重初始化函数,方法来自 Kaiming He,X
2222
- **fan_in** (float16|float32,可选) - 可训练的 Tensor 的 in_features 值。如果设置为 None,程序会自动计算该值。如果你不想使用 in_features,你可以自己设置这个值。默认值为 None。
2323
- **negative_slope** (float,可选) - 只适用于使用 leaky_relu 作为激活函数时的 negative_slope 参数。默认值为 :math:`0.0`。
2424
- **nonlinearity** (str,可选) - 非线性激活函数。默认值为 relu。
25+
- **mode** (str,可选):初始化的模式,参数为 'fan_in' 或 'fan_out'。当设置为 'fan_in' 时,将使用 fan_in 参数进行初始化。当设置为 'fan_out' 时,将使用可训练 Tensor 的 out_features 值进行初始化。默认为 'fan_in'。
2526

2627
.. note::
2728

docs/api/paddle/nn/initializer/KaimingUniform_cn.rst

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
KaimingUniform
44
-------------------------------
55

6-
.. py:class:: paddle.nn.initializer.KaimingUniform(fan_in=None, negative_slope=0.0, nonlinearity='relu')
6+
.. py:class:: paddle.nn.initializer.KaimingUniform(fan_in=None, negative_slope=0.0, nonlinearity='relu', mode='fan_in')
77
88
99
@@ -22,6 +22,7 @@ Kaiming 均匀分布方式的权重初始化函数,方法来自 Kaiming He,X
2222
- **fan_in** (float16|float32,可选) - 可训练的 Tensor 的 in_features 值。如果设置为 None,程序会自动计算该值。如果你不想使用 in_features,你可以自己设置这个值。默认值为 None。
2323
- **negative_slope** (float,可选) - 只适用于使用 leaky_relu 作为激活函数时的 negative_slope 参数。默认值为 :math:`0.0`。
2424
- **nonlinearity** (str,可选) - 非线性激活函数。默认值为 relu。
25+
- **mode** (str,可选):初始化的模式,参数为 'fan_in' 或 'fan_out'。当设置为 'fan_in' 时,将使用 fan_in 参数进行初始化。当设置为 'fan_out' 时,将使用可训练 Tensor 的 out_features 值进行初始化。默认为 'fan_in'。
2526

2627
.. note::
2728

docs/guides/model_convert/convert_from_pytorch/api_difference/init/torch.nn.init.kaiming_normal_.md

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,8 @@ torch.nn.init.kaiming_normal_(tensor,
1515
```python
1616
paddle.nn.initializer.KaimingNormal(fan_in=None,
1717
negative_slope=0.0,
18-
nonlinearity='relu')
18+
nonlinearity='relu'
19+
mode='fan_in')
1920
```
2021

2122
两者用法不同:torch 是 inplace 的用法,paddle 是类设置的,具体如下:
@@ -27,7 +28,7 @@ paddle.nn.initializer.KaimingNormal(fan_in=None,
2728
| tensor | - | n 维 tensor。Paddle 无此参数,因为是通过调用类的 __call__ 函数来进行 tensor 的初始化。 |
2829
| a | negative_slope | 只适用于使用 leaky_relu 作为激活函数时的 negative_slope 参数。仅参数名不一致。 |
2930
| nonlinearity | nonlinearity | 非线性激活函数。参数默认值不一样,PyTorch 默认值为`leaky_relu`,Paddle 默认值为`relu`,Paddle 需保持与 PyTorch 一致。 |
30-
| mode | - | "fan_in"(默认)或 "fan_out"。Paddle 无此参数,mode="fan_out"时,Paddle 无此参数,暂无转写方式|
31+
| mode | mode | "fan_in"(默认)或 "fan_out"。参数完全一致|
3132
| generator | - | 用于采样的伪随机数生成器,Paddle 无此参数,一般对网络训练结果影响不大,可直接删除。 |
3233
| - | fan_in | 可训练的 Tensor 的 in_features 值。PyTorch 无此参数,Paddle 保持默认即可。 |
3334

docs/guides/model_convert/convert_from_pytorch/api_difference/init/torch.nn.init.kaiming_uniform_.md

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,8 @@ torch.nn.init.kaiming_uniform_(tensor,
1414
```python
1515
paddle.nn.initializer.KaimingUniform(fan_in=None,
1616
negative_slope=0.0,
17-
nonlinearity='relu')
17+
nonlinearity='relu',
18+
mode='fan_in')
1819
```
1920

2021
两者用法不同:torch 是 inplace 的用法,paddle 是类设置的,具体如下:
@@ -26,7 +27,7 @@ paddle.nn.initializer.KaimingUniform(fan_in=None,
2627
| tensor | - | n 维 tensor。Paddle 无此参数,因为是通过调用类的 __call__ 函数来进行 tensor 的初始化。 |
2728
| a | negative_slope | 只适用于使用 leaky_relu 作为激活函数时的 negative_slope 参数。仅参数名不一致。 |
2829
| nonlinearity | nonlinearity | 非线性激活函数。参数默认值不一样,PyTorch 默认值为`leaky_relu`,Paddle 默认值为`relu`,Paddle 需保持与 PyTorch 一致。 |
29-
| mode | - | "fan_in"(默认)或 "fan_out"。Paddle 无此参数,mode="fan_out"时,Paddle 无此参数,暂无转写方式|
30+
| mode | mode | "fan_in"(默认)或 "fan_out"。参数完全一致|
3031
| - | fan_in | 可训练的 Tensor 的 in_features 值。PyTorch 无此参数,Paddle 保持默认即可。 |
3132

3233
### 转写示例

0 commit comments

Comments
 (0)