From b04c6fb44008e865263272d71f858ddd9864aee9 Mon Sep 17 00:00:00 2001 From: Double_V Date: Tue, 7 Dec 2021 18:16:13 +0800 Subject: [PATCH] add no_grad --- tools/infer_rec.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/infer_rec.py b/tools/infer_rec.py index cf49348f..36f570b9 100755 --- a/tools/infer_rec.py +++ b/tools/infer_rec.py @@ -37,7 +37,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