Skip to content
Snippets Groups Projects
Commit 71a06b4f authored by Olivier BICHLER's avatar Olivier BICHLER
Browse files

Added missing check

parent 5c93e990
No related branches found
No related tags found
No related merge requests found
......@@ -397,7 +397,10 @@ void normalizeParameters(std::shared_ptr<GraphView> graphView)
if (nodeHasBias)
{
std::shared_ptr<Tensor> biasTensor = getBiasTensor(node);
rescaleTensor(biasTensor, accumulatedRatios[node->name()] );
// Check that a bias is present (as it is optional)
if (biasTensor) {
rescaleTensor(biasTensor, accumulatedRatios[node->name()] );
}
}
}
......
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