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

edit normalizeActivations()

parent b6f593f9
No related branches found
No related tags found
2 merge requests!54Update 0.3.1 -> 0.4.0,!50Enhancement : Quantizer only PTQ
Pipeline #68363 failed
......@@ -818,11 +818,9 @@ void normalizeActivations(std::shared_ptr<GraphView> graphView, std::unordered_m
// If prevNode is Affine, fix the bias ...
if (isAffine(prevNode)) {
if (nodeHasBias(prevNode)) {
insertScalingBelowProducer(prevNode->getParent(2), 1.0 / prevRatio, graphView);
}
}
if (isAffine(prevNode))
if (nodeHasBias(prevNode))
multiplyScalingFactor(prevNode->getParent(2), 1.0 / prevRatio);
}
// Merging nodes handling : use a maximum arbritration ...
......@@ -873,7 +871,6 @@ void normalizeActivations(std::shared_ptr<GraphView> graphView, std::unordered_m
multiplyScalingFactor(prevScalingNode, prevRatio);
}
}
}
......
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