未验证 提交 5784b811 编写于 作者: S SunAhong1993 提交者: GitHub

Merge pull request #29 from PaddlePaddle/develop

merge
...@@ -417,15 +417,15 @@ class CaffeOpMapper(OpMapper): ...@@ -417,15 +417,15 @@ class CaffeOpMapper(OpMapper):
assert params.local_size % 2 == 1 assert params.local_size % 2 == 1
alpha = params.alpha / float(params.local_size) alpha = params.alpha / float(params.local_size)
layer_attrs = { layer_attrs = {
"size": params.local_size, "n": params.local_size,
"k": params.k, "k": params.k,
"alpha": alpha, "alpha": alpha,
"beta": params.beta "beta": params.beta,
} }
self.paddle_graph.add_layer( self.paddle_graph.add_layer(
"paddle.nn.LocalResponseNorm", "paddle.fluid.layers.lrn",
inputs={"input": input.name}, inputs={"input": input.name},
outputs=layer_outputs, outputs=[node.layer_name],
**layer_attrs) **layer_attrs)
......
...@@ -1793,7 +1793,7 @@ class OpSet9(): ...@@ -1793,7 +1793,7 @@ class OpSet9():
inputs_dict = {'x': val_x if isinstance(val_x, str) else val_x.name} inputs_dict = {'x': val_x if isinstance(val_x, str) else val_x.name}
layer_attrs = { layer_attrs = {
"in_channels": num_in_channels, "in_channels": num_in_channels,
"out_channels": num_out_channels, "out_channels": num_out_channels * num_groups,
"kernel_size": kernel_shape, "kernel_size": kernel_shape,
"stride": strides, "stride": strides,
"dilation": dilations, "dilation": dilations,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册