From b71bd9c29879dc62ba1e663e17bc334a1112f405 Mon Sep 17 00:00:00 2001
From: bhalimi <benjamin.halimi@cea.fr>
Date: Wed, 27 Nov 2024 09:58:46 +0000
Subject: [PATCH] add the training flag in the unit tests
---
unit_tests/graph/Test_Matching.cpp | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/unit_tests/graph/Test_Matching.cpp b/unit_tests/graph/Test_Matching.cpp
index 8c5fa222a..ce454c409 100644
--- a/unit_tests/graph/Test_Matching.cpp
+++ b/unit_tests/graph/Test_Matching.cpp
@@ -352,11 +352,11 @@ TEST_CASE("[core/graph] Matching") {
auto g2 = Sequential({
Producer({16, 3, 512, 512}, "dataProvider"),
Conv(3, 4, {5, 5}, "conv1"),
- BatchNorm<2>(4, 1.0e-5, 0.1, "bn1"),
+ BatchNorm<2>(4, 1.0e-5, 0.1, 0, "bn1"),
Conv(4, 4, {5, 5}, "conv2"),
ReLU("relu2"),
Conv(4, 4, {5, 5}, "conv3"),
- BatchNorm<2>(4, 1.0e-5, 0.1, "bn3"),
+ BatchNorm<2>(4, 1.0e-5, 0.1, 0, "bn3"),
FC(4, 4, false, "fc1"),
FC(4, 4, false, "fc2"),
FC(4, 4, false, "fc3"),
--
GitLab