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):
output_a_shape == input_b_shape
), "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
for n in pre_model.graph.node:
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