add no_grad

b8b6f2f0 · Double_V · GitHub · 09604c38 · b8b6f2f0
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

tools/infer_det.py tools/infer_det.py +1 -1

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

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