Merge pull request #165 from mamingjie-China/develop

fix the bug

Merge pull request #165 from mamingjie-China/develop
fix the bug
8a4b80e9 · Jason · GitHub · dc8926c7 · f1f51579 · 8a4b80e9
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

x2paddle/op_mapper/tf_op_mapper.py x2paddle/op_mapper/tf_op_mapper.py +1 -1

x2paddle/op_mapper/tf_op_mapper_nhwc.py x2paddle/op_mapper/tf_op_mapper_nhwc.py +1 -1

未找到文件。
--- a/x2paddle/op_mapper/tf_op_mapper.py
+++ b/x2paddle/op_mapper/tf_op_mapper.py
@@ -1010,7 +1010,7 @@ class TFOpMapper(OpMapper):
        attr = {
            "bias_attr": False,
            "param_attr": string(kernel.layer_name),
-            "num_filters": k_size[3],
+            "num_filters": k_size[2],
            "filter_size": k_size[0:2],
            "stride": strides[2:4],
            "dilation": dilations[2:4],

--- a/x2paddle/op_mapper/tf_op_mapper_nhwc.py
+++ b/x2paddle/op_mapper/tf_op_mapper_nhwc.py
@@ -1007,7 +1007,7 @@ class TFOpMapperNHWC(OpMapper):
        attr = {
            "bias_attr": False,
            "param_attr": string(kernel.layer_name),
-            "num_filters": k_size[3],
+            "num_filters": k_size[2],
            "filter_size": k_size[0:2],
            "stride": strides[2:4],
            "dilation": dilations[2:4],