Fix typo in batch norm bias initialization (#7449)

ea782e38 · Abhinav Arora · GitHub · a9dbdab5 · ea782e38
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

python/paddle/v2/fluid/layers/nn.py python/paddle/v2/fluid/layers/nn.py +1 -1

未找到文件。
--- a/python/paddle/v2/fluid/layers/nn.py
+++ b/python/paddle/v2/fluid/layers/nn.py
@@ -983,7 +983,7 @@ def batch_norm(input,
        default_initializer=Constant(1.0))

    bias = helper.create_parameter(
-        attr=helper.param_attr, shape=param_shape, dtype=dtype, is_bias=True)
+        attr=helper.bias_attr, shape=param_shape, dtype=dtype, is_bias=True)

    mean = helper.create_global_variable(
        dtype=input.dtype,