fix analysis_tester bug

test=develop
6 years ago · 23bdd0a223
parent 875a07c32d
commit 23bdd0a223
1 changed files with 2 additions and 2 deletions
--- a/paddle/fluid/inference/analysis/analyzer_tester.cc
+++ b/paddle/fluid/inference/analysis/analyzer_tester.cc
@ -80,8 +80,8 @@ void TestWord2vecPrediction(const std::string& model_path) {
       i++) {
    LOG(INFO) << "data: " << static_cast<float*>(outputs.front().data.data())[i]
              << " result: " << result[i];
-    PADDLE_ENFORCE(static_cast<float*>(outputs.front().data.data())[i],
-                   result[i]);
+    EXPECT_NEAR(static_cast<float*>(outputs.front().data.data())[i], result[i],
+                1e-3);
  }
 }