add no_grad

2026-10-25 10:17:04 +08:00 · 2021-11-26 07:28:50 +00:00
parent bccba5578f
commit 17a4642103
1 changed files with 1 additions and 0 deletions
@@ -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']