提交 353f006d 编写于 作者: S SunAhong1993

fix

上级 5eba177e
......@@ -417,15 +417,15 @@ class CaffeOpMapper(OpMapper):
assert params.local_size % 2 == 1
alpha = params.alpha / float(params.local_size)
layer_attrs = {
"size": params.local_size,
"k": params.k,
"n": params.local_size,
"k": params.k,
"alpha": alpha,
"beta": params.beta
"beta": params.beta,
}
self.paddle_graph.add_layer(
"paddle.nn.LocalResponseNorm",
"paddle.fluid.layers.lrn",
inputs={"input": input.name},
outputs=layer_outputs,
outputs=[node.layer_name],
**layer_attrs)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册