diff --git a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_lac.py b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_lac.py index d8cb3854d3e23a422377d03effbd6a69aa9d45a9..c9bc8cc647df3374da8ad776bad7abc8a39f132b 100644 --- a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_lac.py +++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_lac.py @@ -28,8 +28,6 @@ from paddle.fluid.dygraph import Embedding, Linear, GRUUnit from paddle.fluid.dygraph import declarative, ProgramTranslator from paddle.fluid.dygraph.io import INFER_MODEL_SUFFIX, INFER_PARAMS_SUFFIX -from predictor_utils import PredictorTools - SEED = 2020 program_translator = ProgramTranslator() @@ -540,7 +538,6 @@ class TestLACModel(unittest.TestCase): dy_pre = self.predict_dygraph(batch) st_pre = self.predict_static(batch) dy_jit_pre = self.predict_dygraph_jit(batch) - predictor_pre = self.predict_analysis_inference(batch) self.assertTrue( np.allclose(dy_pre, st_pre), msg="dy_pre:\n {}\n, st_pre: \n{}.".format(dy_pre, st_pre)) @@ -548,10 +545,6 @@ class TestLACModel(unittest.TestCase): np.allclose(dy_jit_pre, st_pre), msg="dy_jit_pre:\n {}\n, st_pre: \n{}.".format(dy_jit_pre, st_pre)) - self.assertTrue( - np.allclose(predictor_pre, st_pre), - msg="predictor_pre:\n {}\n, st_pre: \n{}.".format(predictor_pre, - st_pre)) def predict_dygraph(self, batch): words, targets, length = batch @@ -602,15 +595,6 @@ class TestLACModel(unittest.TestCase): return pred_res.numpy() - def predict_analysis_inference(self, batch): - words, targets, length = batch - - output = PredictorTools(self.args.model_save_dir, - self.args.model_filename, - self.args.params_filename, [words, length]) - out = output() - return out - if __name__ == "__main__": unittest.main()