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

fix the checkArchitecture

parent c30398cd
No related branches found
No related tags found
1 merge request!5Minor Changes
......@@ -82,8 +82,8 @@ bool checkArchitecture(std::shared_ptr<GraphView> graphView)
for (std::shared_ptr<Node> node : graphView->getNodes())
{
bool isOther = otherNodeTypes.find(node->type()) == otherNodeTypes.end();
if (isOther || isAffine(node) || isSeamless(node)) {
bool isOther = otherNodeTypes.find(node->type()) != otherNodeTypes.end();
if (!isOther && !isAffine(node) && !isSeamless(node)) {
Log::info(" GraphView can't be quantized : node type {} is not supported !", node->type());
return false;
}
......@@ -613,7 +613,7 @@ void quantizeNormalizedNetwork(std::shared_ptr<GraphView> graphView, std::uint8_
void quantizeNetwork(std::shared_ptr<GraphView> graphView, std::uint8_t nbBits, std::vector<std::shared_ptr<Tensor>> inputDataSet)
{
Log::info(" === QUANT PTQ 0.2.5 === ");
Log::info(" === QUANT PTQ 0.2.7 === ");
if (!checkArchitecture(graphView))
return;
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment