From 4edb7c6f34f28aea5498e5afebbf1130e49f8a4e Mon Sep 17 00:00:00 2001
From: Zhibin Li <zhibin.li@intel.com>
Date: Tue, 16 Nov 2021 12:49:50 +0800
Subject: [PATCH] NMS: change back threshold for comparison (#8553)

---
 .../src/single_layer/non_max_suppression.cpp                  | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/inference-engine/tests/functional/shared_test_classes/src/single_layer/non_max_suppression.cpp b/inference-engine/tests/functional/shared_test_classes/src/single_layer/non_max_suppression.cpp
index 2a8d8f8497e..8c477178a25 100644
--- a/inference-engine/tests/functional/shared_test_classes/src/single_layer/non_max_suppression.cpp
+++ b/inference-engine/tests/functional/shared_test_classes/src/single_layer/non_max_suppression.cpp
@@ -91,12 +91,12 @@ void NmsLayerTest::CompareBuffer(const std::vector<std::pair<ngraph::element::Ty
                     case ngraph::element::Type_t::f32:
                         LayerTestsUtils::LayerTestsCommon::Compare(
                                 reinterpret_cast<const float *>(expectedBuffer),
-                                reinterpret_cast<const float *>(actualBuffer), size, 0);
+                                reinterpret_cast<const float *>(actualBuffer), size, threshold);
                         break;
                     case ngraph::element::Type_t::f64:
                         LayerTestsUtils::LayerTestsCommon::Compare(
                                 reinterpret_cast<const double *>(expectedBuffer),
-                                reinterpret_cast<const float *>(actualBuffer), size, 0);
+                                reinterpret_cast<const float *>(actualBuffer), size, threshold);
                         break;
                     default:
                         break;