diff --git a/tools/infer_cls.py b/tools/infer_cls.py
index a588cab433442695e3bd395da63e35a2052de501..36b3c5b51e49cf7ab98d0549fe9fa3f2d40a92f8 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']