From 2307baf5b0d0721b0ebd96ca403576879ec0b168 Mon Sep 17 00:00:00 2001 From: Dun Date: Thu, 21 Feb 2019 11:26:02 +0800 Subject: [PATCH] cherry-pick pr15727 (#15833) * fix group_norm (#15727) --- python/paddle/fluid/layers/nn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/fluid/layers/nn.py b/python/paddle/fluid/layers/nn.py index 46ce58fd2d..586eac7fd6 100644 --- 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", -- GitLab