Skip to content
Snippets Groups Projects
Commit 225ce643 authored by auphelia's avatar auphelia
Browse files

[Transform] Remove output tensor from value_info in insert_hook transform

parent 0b322384
No related branches found
No related tags found
No related merge requests found
......@@ -122,6 +122,7 @@ class InsertHook(Transformation):
else:
model.graph.output.pop()
model.graph.output.append(chk_otensor)
model.graph.value_info.remove(chk_otensor)
model = model.transform(GiveUniqueNodeNames())
model = model.transform(GiveReadableTensorNames())
graph_modified = True
......
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