未验证 提交 8a4b80e9 编写于 作者: J Jason 提交者: GitHub

Merge pull request #165 from mamingjie-China/develop

fix the bug
...@@ -1010,7 +1010,7 @@ class TFOpMapper(OpMapper): ...@@ -1010,7 +1010,7 @@ class TFOpMapper(OpMapper):
attr = { attr = {
"bias_attr": False, "bias_attr": False,
"param_attr": string(kernel.layer_name), "param_attr": string(kernel.layer_name),
"num_filters": k_size[3], "num_filters": k_size[2],
"filter_size": k_size[0:2], "filter_size": k_size[0:2],
"stride": strides[2:4], "stride": strides[2:4],
"dilation": dilations[2:4], "dilation": dilations[2:4],
......
...@@ -1007,7 +1007,7 @@ class TFOpMapperNHWC(OpMapper): ...@@ -1007,7 +1007,7 @@ class TFOpMapperNHWC(OpMapper):
attr = { attr = {
"bias_attr": False, "bias_attr": False,
"param_attr": string(kernel.layer_name), "param_attr": string(kernel.layer_name),
"num_filters": k_size[3], "num_filters": k_size[2],
"filter_size": k_size[0:2], "filter_size": k_size[0:2],
"stride": strides[2:4], "stride": strides[2:4],
"dilation": dilations[2:4], "dilation": dilations[2:4],
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册