From 2128f5f79cdde938e27812474a1f3f7d3f4a568e Mon Sep 17 00:00:00 2001 From: channingss Date: Thu, 6 Feb 2020 20:35:04 +0800 Subject: [PATCH] [ONNX]fix bug for InstanceNormalization --- .../onnx_custom_layer/InstanceNormalization.py | 13 +++++-------- 1 file changed, 5 insertions(+), 8 deletions(-) diff --git a/x2paddle/op_mapper/onnx_custom_layer/InstanceNormalization.py b/x2paddle/op_mapper/onnx_custom_layer/InstanceNormalization.py index 0d767d7..db5e208 100644 --- a/x2paddle/op_mapper/onnx_custom_layer/InstanceNormalization.py +++ b/x2paddle/op_mapper/onnx_custom_layer/InstanceNormalization.py @@ -30,16 +30,13 @@ def InstanceNormalization_layer(inputs, name=None): if name is not None: scale_name = name + "_scale" offset_name = name + "_offset" - scale_param = fluid.ParamAttr(name=scale_name, - initializer=fluid.initializer.Constant(1.0), - trainable=True) - offset_param = fluid.ParamAttr(name=offset_name, - initializer=fluid.initializer.Constant(0.0), - trainable=True) - scale = fluid.layers.create_parameter(attr=scale_param, + + scale_param = inputs[1] + offset_param = inputs[2] + scale = fluid.layers.create_parameter(name=scale_param.name, shape=input_.shape[1:2], dtype="float32") - offset = fluid.layers.create_parameter(attr=offset_param, + offset = fluid.layers.create_parameter(name=offset_param.name, shape=input_.shape[1:2], dtype="float32") -- GitLab