From 3177760f515b602b224094e72f62eb82ba0e902d Mon Sep 17 00:00:00 2001 From: Double_V 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