From 1a166e590fdd6570c316e1a76c836fd0617c172e Mon Sep 17 00:00:00 2001
From: auphelia <jakobapk@web.de>
Date: Fri, 5 Jun 2020 15:44:19 +0100
Subject: [PATCH] [Test] Commit quantized avg pool brevitas export after
pre-commit
---
tests/brevitas/test_brevitas_avg_pool_export.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/tests/brevitas/test_brevitas_avg_pool_export.py b/tests/brevitas/test_brevitas_avg_pool_export.py
index 17b2bc5aa..3b9929b09 100644
--- a/tests/brevitas/test_brevitas_avg_pool_export.py
+++ b/tests/brevitas/test_brevitas_avg_pool_export.py
@@ -64,9 +64,9 @@ def test_brevitas_avg_pool_export(kernel_size, stride, signed, bit_width):
)
b_avgpool.eval()
expected = b_avgpool.forward(input_quant_tensor).tensor.detach().numpy()
-
+
# finn execution
- idict = {model.graph.input[0].name : inp}
+ idict = {model.graph.input[0].name: inp}
odict = oxe.execute_onnx(model, idict, True)
produced = odict[model.graph.output[0].name]
--
GitLab