From 7bddcc80f9386e5ee2a3460ae34973a88aeca7c0 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Fri, 8 May 2020 15:48:57 +0100
Subject: [PATCH] [Test] don't save intermediate onnx models in streamline_cnv
---
tests/transformation/streamline/test_streamline_cnv.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/tests/transformation/streamline/test_streamline_cnv.py b/tests/transformation/streamline/test_streamline_cnv.py
index cfa5abe58..56dcd2607 100644
--- a/tests/transformation/streamline/test_streamline_cnv.py
+++ b/tests/transformation/streamline/test_streamline_cnv.py
@@ -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()
--
GitLab