From 759cb23711c0dab4d52dfb45ccc275602fa999fd Mon Sep 17 00:00:00 2001 From: Xin Pan Date: Thu, 31 May 2018 14:27:05 +0800 Subject: [PATCH] follow comments --- paddle/contrib/inference/test_paddle_inference_api_impl.cc | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/paddle/contrib/inference/test_paddle_inference_api_impl.cc b/paddle/contrib/inference/test_paddle_inference_api_impl.cc index 09ca1832db6..57c39c50f1e 100644 --- a/paddle/contrib/inference/test_paddle_inference_api_impl.cc +++ b/paddle/contrib/inference/test_paddle_inference_api_impl.cc @@ -144,8 +144,7 @@ TEST(paddle_inference_api_impl, image_classification) { float* data = static_cast(outputs[0].data.data); float* lod_data = output1.data(); for (size_t j = 0; j < len / sizeof(float); ++j) { - EXPECT_LT(lod_data[j] - data[j], 1e-3); - EXPECT_GT(lod_data[j] - data[j], -1e-3); + EXPECT_NEAR (lod_data[j], data[j], 1e-6); } free(data); } -- GitLab