From 1ead5e0a0ddcbf4a3b4d7db259f4aa8e6f10129c Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Fri, 11 Sep 2020 13:07:05 +0200
Subject: [PATCH] [Test] change k settings for test_topk_insert
several channels getting out the same value, causing different
topk outputs from npy and finn
---
tests/transformation/test_topk_insert.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/tests/transformation/test_topk_insert.py b/tests/transformation/test_topk_insert.py
index a18e63384..a9faac4df 100644
--- a/tests/transformation/test_topk_insert.py
+++ b/tests/transformation/test_topk_insert.py
@@ -21,7 +21,7 @@ import pytest
export_onnx_path = "test_topk_insert.onnx"
-@pytest.mark.parametrize("k", [1, 5, 10])
+@pytest.mark.parametrize("k", [1, 2])
def test_topk_insert(k):
tfc = get_test_model_trained("TFC", 1, 1)
bo.export_finn_onnx(tfc, (1, 1, 28, 28), export_onnx_path)
--
GitLab