Merge pull request #11071 from panyx0718/fix_infer_test

make infer test more stable.
8 years ago · f437c46ffd
parent d90610624f 44ea234343
commit f437c46ffd
1 changed files with 1 additions and 2 deletions
--- 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<float*>(outputs[0].data.data);
  float* lod_data = output1.data<float>();
  for (size_t j = 0; j < len / sizeof(float); ++j) {
-    EXPECT_LT(lod_data[j] - data[j], 1e-10);
-    EXPECT_GT(lod_data[j] - data[j], -1e-10);
+    EXPECT_NEAR(lod_data[j], data[j], 1e-3);
  }
  free(data);
 }