Skip to content

GitLab

Unverified Commit 681ccc59 authored by Ningxin Zheng's avatar Ningxin Zheng Committed by GitHub
Browse files

remove pdb (#4151)

parent 396ae65c
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 6 deletions
nni/retiarii/converter/graph_gen.py
View file @ 681ccc59
......@@ -818,12 +818,10 @@ class GraphConverterWithShape(GraphConverter):
graph.del_edge(out_edge)
for output_node_edge in node_graph.output_node.incoming_edges:
if output_node_edge.head_slot == out_edge.tail_slot:
try:
graph.add_edge(
head=(id_to_new_node[output_node_edge.head.id], output_node_edge.head_slot),
tail=(out_edge.tail, out_edge.tail_slot))
except:
import pdb; pdb.set_trace()
graph.add_edge(
head=(id_to_new_node[output_node_edge.head.id], output_node_edge.head_slot),
tail=(out_edge.tail, out_edge.tail_slot))
for edge in node_graph.edges:
if edge.head == node_graph.input_node or edge.tail == node_graph.output_node:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment