diff options
| -rw-r--r-- | src/armnnUtils/ModelAccuracyChecker.hpp | 11 |
1 files changed, 7 insertions, 4 deletions
diff --git a/src/armnnUtils/ModelAccuracyChecker.hpp b/src/armnnUtils/ModelAccuracyChecker.hpp index abf994b5e1..cdd2af0ac5 100644 --- a/src/armnnUtils/ModelAccuracyChecker.hpp +++ b/src/armnnUtils/ModelAccuracyChecker.hpp @@ -70,13 +70,16 @@ public: unsigned int index = 1; for (std::pair<int, float> element : setOfPredictions) { - if(element.first == value) + if (index >= m_TopK.size()) { - ++m_TopK[index]; - } else + break; + } + if (element.first == value) { - ++index; + ++m_TopK[index]; + break; } + ++index; } } |