提交 32aa07a9 编写于 作者: W wjj19950828

fixed for ci

上级 07929a13
......@@ -20,7 +20,7 @@ class Normalize(object):
self.axis = axis
def __call__(self, x, param):
l2_norm = paddle.norm(x=x, p=2, axis=1)
l2_norm = paddle.norm(x=x, p=2, axis=1, keepdim=True)
param = paddle.reshape(param, [param.shape[-1]])
perm = list(range(len(l2_norm.shape)))
perm.pop(self.axis)
......
......@@ -429,10 +429,10 @@ class CaffeOpMapper():
assert params.local_size % 2 == 1
alpha = params.alpha / float(params.local_size)
layer_attrs = {
"n": params.local_size,
"k": params.k,
"size": params.local_size,
"alpha": alpha,
"beta": params.beta,
"k": params.k,
}
self.paddle_graph.add_layer(
"paddle.nn.LocalResponseNorm",
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册