From 24a42dc8843476be781e48c26db62b0bac87c209 Mon Sep 17 00:00:00 2001
From: Christophe Guillon <christophe.guillon@inria.fr>
Date: Tue, 15 Oct 2024 13:01:38 +0200
Subject: [PATCH] [Tests] Reduce export test problem size
Reduce parameter sizes for export test to reduce test time.
---
aidge_core/unit_tests/test_export.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/aidge_core/unit_tests/test_export.py b/aidge_core/unit_tests/test_export.py
index 7b8620d60..b8e1f0ba9 100644
--- a/aidge_core/unit_tests/test_export.py
+++ b/aidge_core/unit_tests/test_export.py
@@ -78,14 +78,14 @@ class test_export(unittest.TestCase):
model = aidge_core.sequential(
[
aidge_core.FC(
- in_channels=32 * 32 * 3, out_channels=512, name="InputNode"
+ in_channels=32 * 32 * 3, out_channels=64, name="InputNode"
),
aidge_core.ReLU(name="Relu0"),
- aidge_core.FC(in_channels=512, out_channels=256, name="FC1"),
+ aidge_core.FC(in_channels=64, out_channels=32, name="FC1"),
aidge_core.ReLU(name="Relu1"),
- aidge_core.FC(in_channels=256, out_channels=128, name="FC2"),
+ aidge_core.FC(in_channels=32, out_channels=16, name="FC2"),
aidge_core.ReLU(name="Relu2"),
- aidge_core.FC(in_channels=128, out_channels=10, name="OutputNode"),
+ aidge_core.FC(in_channels=16, out_channels=10, name="OutputNode"),
]
)
--
GitLab