Skip to content
Snippets Groups Projects
Commit 9cd88c39 authored by Yaman Umuroglu's avatar Yaman Umuroglu
Browse files

[Transform] don't save internals in MergeONNXModels

parent de283eae
No related branches found
No related tags found
No related merge requests found
...@@ -106,9 +106,6 @@ class MergeONNXModels(Transformation): ...@@ -106,9 +106,6 @@ class MergeONNXModels(Transformation):
output_a_shape == input_b_shape output_a_shape == input_b_shape
), "Models can't be merged! Shapes don't match." ), "Models can't be merged! Shapes don't match."
pre_model.save("pre.onnx")
post_model.save("post.onnx")
# connect output of one model to input of the other # connect output of one model to input of the other
for n in pre_model.graph.node: for n in pre_model.graph.node:
if output_model_a == n.output[0]: if output_model_a == n.output[0]:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment