Skip to content
Snippets Groups Projects
Commit 266b8ed0 authored by Noam Zerah's avatar Noam Zerah
Browse files

Fix Batchnorm arguments in QuantRecipes*

parent 1f5ee473
No related branches found
No related tags found
2 merge requests!24Adding MulCompensation Nodes to the PTQ pipeline,!22Refactor: removing scaling nodes
Pipeline #60933 passed
......@@ -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, batchnormNodeName);
std::shared_ptr<Node> batchnormNode = BatchNorm<2>(nb_channels, 1e-5, 0.1, false, batchnormNodeName);
batchnormNode->getOperator()->setDataType(DataType::Float32);
batchnormNode->getOperator()->setBackend("cpu");
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment