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

fix

上级 6e600d8f
......@@ -184,7 +184,7 @@ class TFGraph(Graph):
node = super(TFGraph, self).get_node(new_node_name, copy)
if node is None:
return None
if node.layer_type not in ["Unpack", "Split"]:
if node.layer_type == "Switch":
if hasattr(node, 'index'):
del node.index
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.
先完成此消息的编辑!
想要评论请 注册