diff --git a/x2paddle/op_mapper/dygraph/caffe2paddle/caffe_op_mapper.py b/x2paddle/op_mapper/dygraph/caffe2paddle/caffe_op_mapper.py index c89c614c22ee14919fa2d6e543cb431bbd1781b0..d654c5b1f905b293cf20505331573d936e667488 100644 --- a/x2paddle/op_mapper/dygraph/caffe2paddle/caffe_op_mapper.py +++ b/x2paddle/op_mapper/dygraph/caffe2paddle/caffe_op_mapper.py @@ -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) diff --git a/x2paddle/op_mapper/dygraph/onnx2paddle/opset9/opset.py b/x2paddle/op_mapper/dygraph/onnx2paddle/opset9/opset.py index dd2d24d2891bcbbc4d29bd73393d703d10ed0793..d4abf556c7dc28be28ecb1317bd2842c903b43ab 100755 --- a/x2paddle/op_mapper/dygraph/onnx2paddle/opset9/opset.py +++ b/x2paddle/op_mapper/dygraph/onnx2paddle/opset9/opset.py @@ -1793,7 +1793,7 @@ class OpSet9(): inputs_dict = {'x': val_x if isinstance(val_x, str) else val_x.name} layer_attrs = { "in_channels": num_in_channels, - "out_channels": num_out_channels, + "out_channels": num_out_channels * num_groups, "kernel_size": kernel_shape, "stride": strides, "dilation": dilations, diff --git a/x2paddle/tests/__init__.py b/x2paddle/tests/__init__.py deleted file mode 100644 index e69de29bb2d1d6434b8b29ae775ad8c2e48c5391..0000000000000000000000000000000000000000