diff --git a/x2paddle/op_mapper/caffe_custom_layer/normalize.py b/x2paddle/op_mapper/caffe_custom_layer/normalize.py index 15bbb2043a1e385d987c577d099c8db902de3f26..19c583acc7ab73cd63d0bc7f373488e9437aec4c 100644 --- a/x2paddle/op_mapper/caffe_custom_layer/normalize.py +++ b/x2paddle/op_mapper/caffe_custom_layer/normalize.py @@ -17,7 +17,7 @@ def normalize_layer(inputs, scale_param = fluid.layers.create_parameter( shape=[1] if channel_shared else [1, 1, 1, input_shape[0][1]], dtype=input.dtype, - attr=name + '_scale') + attr=fluid.ParamAttr(name=name + '_scale')) scale_param = fluid.layers.reshape(x=scale_param, \ shape=[1] if channel_shared else [input_shape[0][1]]) out = fluid.layers.elementwise_mul(