Skip to content
Snippets Groups Projects
Commit 88f4bd71 authored by Benjamin Halimi's avatar Benjamin Halimi
Browse files

remove the bn train flag

parent 4fa3e9a7
No related branches found
No related tags found
2 merge requests!24Adding MulCompensation Nodes to the PTQ pipeline,!22Refactor: removing scaling nodes
...@@ -62,7 +62,7 @@ void insertBatchNormNodes(std::shared_ptr<GraphView> graphView) ...@@ -62,7 +62,7 @@ void insertBatchNormNodes(std::shared_ptr<GraphView> graphView)
std::cout << " NB CHANNELS = " << nb_channels << std::endl; // TODO : remove this ... std::cout << " NB CHANNELS = " << nb_channels << std::endl; // TODO : remove this ...
std::string batchnormNodeName = makeUniqueName(parentNode->name() + "_BN", graphView); 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()->setDataType(DataType::Float32);
batchnormNode->getOperator()->setBackend("cpu"); 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