From 63dc2f7d3830c86a60f10199c641512f4f2c9270 Mon Sep 17 00:00:00 2001
From: NAUD Maxence <maxence.naud@cea.fr>
Date: Thu, 14 Dec 2023 15:02:40 +0000
Subject: [PATCH] fix bug with batchnorm initialization
---
aidge_backend_cpu/unit_tests/test_recipies.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/aidge_backend_cpu/unit_tests/test_recipies.py b/aidge_backend_cpu/unit_tests/test_recipies.py
index 841c1559..e343fad1 100644
--- a/aidge_backend_cpu/unit_tests/test_recipies.py
+++ b/aidge_backend_cpu/unit_tests/test_recipies.py
@@ -31,7 +31,7 @@ class test_recipies(unittest.TestCase):
input_node = aidge_core.Producer(input_tensor, "X")
conv = aidge_core.Conv2D(1, 1, [3, 3], name="Conv0")
- bn = aidge_core.BatchNorm2D(name="Add0")
+ bn = aidge_core.BatchNorm2D(1, name="Add0")
graph_view = aidge_core.sequential([conv, bn])
--
GitLab