diff --git a/src/recipes/QuantRecipes.cpp b/src/recipes/QuantRecipes.cpp index 2a93e429d102ba8ab1f54e0d723b816f76fe2ca7..4748f8890ad647e59eb0acf85dfe2e2708e26da7 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");