add no_grad

b04c6fb4 · Double_V · GitHub · b8b6f2f0 · b04c6fb4
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

tools/infer_rec.py tools/infer_rec.py +1 -1

未找到文件。
--- 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']