onnx-graphsurgeon - remove unused input in modified graph in example …

…04 (NVIDIA#2540) * remove unused input from the modified graph in example 04 in onnx-graphsurgeon Signed-off-by: Kelvinthedrugger <[email protected]>
hiennguyen9874 · Dec 20, 2022 · c6eea95 · c6eea95
1 parent 8423367
commit c6eea95
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tools/onnx-graphsurgeon/examples/04_modifying_a_model/modify.py b/tools/onnx-graphsurgeon/examples/04_modifying_a_model/modify.py
@@ -43,6 +43,7 @@
 # Therefore, you should only need to sort the graph when you have added new nodes out-of-order.
 # In this case, the identity node is already in the correct spot (it is the last node,
 # and was appended to the end of the list), but to be on the safer side, we can sort anyway.
-graph.cleanup().toposort()
+
+graph.cleanup(remove_unused_graph_inputs=True).toposort()
 
 onnx.save(gs.export_onnx(graph), "modified.onnx")