From 79f0bb5d63675bfe100946b2db8aee0101500574 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Mon, 10 May 2021 19:17:21 +0100
Subject: [PATCH] [Test] cybsec: add comment re QuantTensor for input
---
tests/end2end/test_end2end_cybsec_mlp.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/tests/end2end/test_end2end_cybsec_mlp.py b/tests/end2end/test_end2end_cybsec_mlp.py
index 962153579..63d6a91e3 100644
--- a/tests/end2end/test_end2end_cybsec_mlp.py
+++ b/tests/end2end/test_end2end_cybsec_mlp.py
@@ -116,7 +116,7 @@ def test_end2end_cybsec_mlp_export():
model_for_export = CybSecMLPForExport(model)
export_onnx_path = get_checkpoint_name("export")
input_shape = (1, 600)
-
+ # create a QuantTensor instance to mark the input as bipolar during export
input_a = np.random.randint(0, 1, size=input_shape).astype(np.float32)
input_a = 2 * input_a - 1
scale = 1.0
--
GitLab