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

[Test] don't save intermediate onnx models in streamline_cnv

parent e16a0187
No related branches found
No related tags found
No related merge requests found
......@@ -71,9 +71,9 @@ def test_streamline_cnv(size, wbits, abits):
input_dict = {"global_in": input_tensor}
expected_ctx = oxe.execute_onnx(model, input_dict, True)
expected = expected_ctx[model.graph.output[0].name]
model.save("orig_cnv.onnx")
# model.save("orig_cnv.onnx")
model = model.transform(Streamline())
model.save("streamlined_cnv.onnx")
# model.save("streamlined_cnv.onnx")
produced_ctx = oxe.execute_onnx(model, input_dict, True)
produced = produced_ctx[model.graph.output[0].name]
assert np.isclose(expected, produced, atol=1e-3).all()
......
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