提交 1cbb878c 编写于 作者: S SunAhong1993

fix

上级 6e600d8f
...@@ -184,7 +184,7 @@ class TFGraph(Graph): ...@@ -184,7 +184,7 @@ class TFGraph(Graph):
node = super(TFGraph, self).get_node(new_node_name, copy) node = super(TFGraph, self).get_node(new_node_name, copy)
if node is None: if node is None:
return None return None
if node.layer_type not in ["Unpack", "Split"]: if node.layer_type == "Switch":
if hasattr(node, 'index'): if hasattr(node, 'index'):
del node.index del node.index
if len(items) == 1 and node.layer_type in self.multi_out_ops: if len(items) == 1 and node.layer_type in self.multi_out_ops:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册