From c2a7e39fdb9cc535df3ba332e88ecb4e96a741ab Mon Sep 17 00:00:00 2001 From: SunAhong1993 Date: Wed, 4 Sep 2019 17:04:21 +0800 Subject: [PATCH] fix the normalize --- x2paddle/op_mapper/caffe_custom_layer/normalize.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/x2paddle/op_mapper/caffe_custom_layer/normalize.py b/x2paddle/op_mapper/caffe_custom_layer/normalize.py index 0aa4c2f..577a674 100644 --- a/x2paddle/op_mapper/caffe_custom_layer/normalize.py +++ b/x2paddle/op_mapper/caffe_custom_layer/normalize.py @@ -18,6 +18,8 @@ def normalize_layer(inputs, shape=[1] if channel_shared else [input_shape[0][1]], dtype=input.dtype, attr=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(x=l2_norm, y=scale_param, axis=-1 if channel_shared else 1) -- GitLab