From 88f4bd7132c68cbde1a74b692bd1b856bcca3abb Mon Sep 17 00:00:00 2001
From: bhalimi <benjamin.halimi@cea.fr>
Date: Wed, 30 Oct 2024 13:56:18 +0000
Subject: [PATCH] remove the bn train 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