From f3245a8112e8deead7d92d56230f00f50595a500 Mon Sep 17 00:00:00 2001
From: Hendrik Borras <hendrikborras@web.de>
Date: Wed, 20 Oct 2021 10:02:11 +0100
Subject: [PATCH] Cleaned up open ToDos.
---
src/finn/transformation/qonnx/infer_quant_avg_pool_2d.py | 2 +-
tests/transformation/test_qonnx_to_finn.py | 4 +---
2 files changed, 2 insertions(+), 4 deletions(-)
diff --git a/src/finn/transformation/qonnx/infer_quant_avg_pool_2d.py b/src/finn/transformation/qonnx/infer_quant_avg_pool_2d.py
index 3de317b9e..f563e885d 100644
--- a/src/finn/transformation/qonnx/infer_quant_avg_pool_2d.py
+++ b/src/finn/transformation/qonnx/infer_quant_avg_pool_2d.py
@@ -162,7 +162,7 @@ class AvgPoolAndTruncToQuantAvgPool(Transformation):
t_inst = getCustomOp(t_node)
signed = t_inst._get_signed_from_upstream(model)
# ToDo: Change this to NHWC,
- # when the channels last input comes around
+ # when the channels last layout comes around.
data_layout = "NCHW"
# Insert scale nodes, QuantAvgPool2d node and required tensors
diff --git a/tests/transformation/test_qonnx_to_finn.py b/tests/transformation/test_qonnx_to_finn.py
index 519b18a28..df7d63e3d 100644
--- a/tests/transformation/test_qonnx_to_finn.py
+++ b/tests/transformation/test_qonnx_to_finn.py
@@ -88,9 +88,7 @@ def analysis_testing_for_no_quant_nodes(model):
return dict()
-# ToDo: Add KWS networks, when they are ready to be added to finn-examples.
-# ToDo: Add RadioML_VGG10, if possible
-# This test currently takes about 4 min and 42 seconds
+# This test currently takes about 4 min and 20 seconds
@pytest.mark.parametrize("abits", [1, 2])
@pytest.mark.parametrize("wbits", [1, 2])
@pytest.mark.parametrize("model_name", ["TFC", "SFC", "LFC", "CNV", "mobilenet"])
--
GitLab