fix group_norm (#15727)

* fix group_norm * test=develop
6 years ago · f7b768d364
parent e4b9fcdbd2
commit f7b768d364
1 changed files with 1 additions and 1 deletions
--- 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",