From 54c3b2b5a955f8965b3e1481b6219571f5aed5dc Mon Sep 17 00:00:00 2001
From: auphelia <jakobapk@web.de>
Date: Thu, 25 Jun 2020 15:38:50 +0100
Subject: [PATCH] [Test] Remove saving points in absorb mul into top k test
---
tests/transformation/test_absorb_mul_into_topk.py | 2 --
1 file changed, 2 deletions(-)
diff --git a/tests/transformation/test_absorb_mul_into_topk.py b/tests/transformation/test_absorb_mul_into_topk.py
index 71af5ddd9..1394220f7 100644
--- a/tests/transformation/test_absorb_mul_into_topk.py
+++ b/tests/transformation/test_absorb_mul_into_topk.py
@@ -77,9 +77,7 @@ def test_absorb_mul_into_topk(mul_positive, scalar):
model = model.transform(InferDataTypes())
model = model.transform(GiveUniqueNodeNames())
model = model.transform(GiveReadableTensorNames())
- model.save("test.onnx")
model_transformed = model.transform(AbsorbScalarMulIntoTopK())
- model_transformed.save("test2.onnx")
# compare execution results
inp_values = np.random.uniform(low=-10, high=10, size=(1, 1, 1, 1000)).astype(
--
GitLab