From 552f5a22e7e68964313e0c74a9f76118d75ccf01 Mon Sep 17 00:00:00 2001
From: Double_V <liuvv0203@163.com>
Date: Tue, 7 Dec 2021 18:17:43 +0800
Subject: [PATCH] add no_grad()

---
 tools/infer_e2e.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tools/infer_e2e.py b/tools/infer_e2e.py
index 1cd468b8..332d4ae9 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']
 
-- 
GitLab