未验证 提交 812bbc5a 编写于 作者: J Jason 提交者: GitHub

Merge pull request #108 from SunAhong1993/develop

add DepthwiseConvolution
......@@ -39,6 +39,8 @@ class CaffeOpMapper(OpMapper):
print("Total nodes: {}".format(len(self.graph.topo_sort)))
for node_name in self.graph.topo_sort:
node = self.graph.get_node(node_name)
if node.layer_type == 'DepthwiseConvolution':
node.layer_type = 'ConvolutionDepthwise'
op = node.layer_type
if hasattr(self, op):
self.set_node_shape(node)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册