diff --git a/tests/onnx/onnxbase.py b/tests/onnx/onnxbase.py index 41770f9f1c6711a902b9e29704448f6434d8fd50..99d5a03c15ab13dcdd365a3fb2084a8567e04463 100644 --- a/tests/onnx/onnxbase.py +++ b/tests/onnx/onnxbase.py @@ -214,6 +214,7 @@ class ONNXConverter(object): paddle.disable_static() # run model = paddle.jit.load(paddle_path) + model.eval() result = model(*paddle_tensor_feed) shutil.rmtree( os.path.join(self.pwd, self.name, self.name + '_' + str(ver) +