diff --git a/tools/infer_e2e.py b/tools/infer_e2e.py
index 1cd468b8e552237af31d985b8b68ddbeecba9c96..332d4ae9b606db1f056f6a2de569c2a40c7cefab 100755
--- a/tools/infer_e2e.py
+++ b/tools/infer_e2e.py
@@ -61,7 +61,7 @@ def draw_e2e_res(dt_boxes, strs, config, img, img_name):
         cv2.imwrite(save_path, src_im)
         logger.info("The e2e Image saved in {}".format(save_path))
 
-
+@paddle.no_grad()
 def main():
     global_config = config['Global']