From 661b40e7e9f4e8c6d9e6dd614e9d1bb54bb77c13 Mon Sep 17 00:00:00 2001
From: cmoineau <cyril.moineau@cea.fr>
Date: Tue, 8 Oct 2024 12:47:38 +0000
Subject: [PATCH] [ExportSerialize] Add forwardDims before running export.
---
aidge_core/unit_tests/test_export.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/aidge_core/unit_tests/test_export.py b/aidge_core/unit_tests/test_export.py
index 23a65c8a2..32d902b5f 100644
--- a/aidge_core/unit_tests/test_export.py
+++ b/aidge_core/unit_tests/test_export.py
@@ -88,6 +88,7 @@ class test_export(unittest.TestCase):
)
initFiller(model)
+ model.forward_dims([[1, 32*32*3]])
# Export model
aidge_core.serialize_to_cpp(self.EXPORT_PATH, model)
--
GitLab