From 17a4642103db8acdeece2860b18497a6457ad282 Mon Sep 17 00:00:00 2001 From: zhiqiu Date: Fri, 26 Nov 2021 06:36:51 +0000 Subject: [PATCH] add no_grad --- tools/infer_det.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tools/infer_det.py b/tools/infer_det.py index bb2cca73..1c679e0f 100755 --- a/tools/infer_det.py +++ b/tools/infer_det.py @@ -53,6 +53,7 @@ def draw_det_res(dt_boxes, config, img, img_name, save_path): logger.info("The detected Image saved in {}".format(save_path)) +@paddle.no_grad() def main(): global_config = config['Global'] -- GitLab