From 3177760f515b602b224094e72f62eb82ba0e902d Mon Sep 17 00:00:00 2001
From: Double_V <liuvv0203@163.com>
Date: Tue, 7 Dec 2021 18:16:58 +0800
Subject: [PATCH] add no_grad

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

diff --git a/tools/infer_cls.py b/tools/infer_cls.py
index a588cab4..36b3c5b5 100755
--- a/tools/infer_cls.py
+++ b/tools/infer_cls.py
@@ -36,7 +36,7 @@ from ppocr.utils.save_load import init_model
 from ppocr.utils.utility import get_image_file_list
 import tools.program as program
 
-
+@paddle.no_grad()
 def main():
     global_config = config['Global']
 
-- 
GitLab