add no_grad

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

Showing with 1 addition and 1 deletion

tools/infer_cls.py tools/infer_cls.py +1 -1

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