From 376c66a3dc72da7480f9712c6891d610477c46a0 Mon Sep 17 00:00:00 2001
From: bhalimi <benjamin.halimi@cea.fr>
Date: Thu, 31 Oct 2024 14:22:24 +0000
Subject: [PATCH] remove the bn training flag
---
src/recipes/QuantRecipes.cpp | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/recipes/QuantRecipes.cpp b/src/recipes/QuantRecipes.cpp
index 2a93e42..4748f88 100644
--- a/src/recipes/QuantRecipes.cpp
+++ b/src/recipes/QuantRecipes.cpp
@@ -62,7 +62,7 @@ void insertBatchNormNodes(std::shared_ptr<GraphView> graphView)
std::cout << " NB CHANNELS = " << nb_channels << std::endl; // TODO : remove this ...
std::string batchnormNodeName = makeUniqueName(parentNode->name() + "_BN", graphView);
- std::shared_ptr<Node> batchnormNode = BatchNorm<2>(nb_channels, 1e-5, 0.1, 0, batchnormNodeName);
+ std::shared_ptr<Node> batchnormNode = BatchNorm<2>(nb_channels, 1e-5, 0.1, /*0,*/ batchnormNodeName);
batchnormNode->getOperator()->setDataType(DataType::Float32);
batchnormNode->getOperator()->setBackend("cpu");
--
GitLab