diff --git a/paddle/contrib/inference/test_paddle_inference_api_impl.cc b/paddle/contrib/inference/test_paddle_inference_api_impl.cc index 09ca1832db619aa99d571eeddc7ef7709620f4c9..57c39c50f1e0f9535d8502ef441056ceec2d78f3 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); }