cherry-pick pr15727 (#15833)

* fix group_norm (#15727)

cherry-pick pr15727 (#15833)
* fix group_norm (#15727)
2307baf5 · Dun · qingqing01 · e5c8a199 · 2307baf5
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

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

未找到文件。
--- a/python/paddle/fluid/layers/nn.py
+++ b/python/paddle/fluid/layers/nn.py
@@ -3236,7 +3236,7 @@ def group_norm(input,
    # create output
    mean_out = helper.create_variable(dtype=dtype, stop_gradient=True)
    variance_out = helper.create_variable(dtype=dtype, stop_gradient=True)
-    group_norm_out = helper.create_variable(dtype)
+    group_norm_out = helper.create_variable(dtype=dtype)

    helper.append_op(
        type="group_norm",