fix group_norm (#15727)

* fix group_norm * test=develop

fix group_norm (#15727)
* fix group_norm * test=develop
f7b768d3 · Dun · GitHub · e4b9fcdb · f7b768d3
显示空白变更内容
内联并排

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",