set_global_initializer

paddle.nn.initializer. set_global_initializer ( weight_init, bias_init=None )

该 API 用于设置 Paddle 框架中全局的参数初始化方法。该 API 只对位于其后的代码生效。

模型参数为模型中的 weight 和 bias 统称,在框架中对应 paddle.ParamAttr 类,继承自 paddle.Tensor,是一种可持久化的 variable。 该 API 的设置仅对模型参数生效,对通过 create_global_varcn_api_paddle_Tensor_create_tensor 等 API 创建的变量不会生效。

如果创建网络层时还通过 param_attrbias_attr 设置了初始化方式,这里的全局设置将不会生效,因为其优先级更低。

参数

  • weight_init (Initializer) - 设置框架的全局的 weight 参数初始化方法。

  • bias_init (Initializer,可选) - 设置框架的全局的 bias 参数初始化方法。默认:None。

返回

代码示例

COPY-FROM: paddle.nn.initializer.set_global_initializer