add no_grad()

552f5a22 · Double_V · GitHub · 3177760f · 552f5a22
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

tools/infer_e2e.py tools/infer_e2e.py +1 -1

未找到文件。
--- a/tools/infer_e2e.py
+++ b/tools/infer_e2e.py
@@ -61,7 +61,7 @@ def draw_e2e_res(dt_boxes, strs, config, img, img_name):
        cv2.imwrite(save_path, src_im)
        logger.info("The e2e Image saved in {}".format(save_path))

-
+@paddle.no_grad()
 def main():
    global_config = config['Global']