未验证 提交 e2755f2c 编写于 作者: D Double_V 提交者: GitHub

Merge pull request #4774 from zhiqiu/dev/fix_infer2

add no_grad for infer function
...@@ -53,6 +53,7 @@ def draw_det_res(dt_boxes, config, img, img_name, save_path): ...@@ -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)) logger.info("The detected Image saved in {}".format(save_path))
@paddle.no_grad()
def main(): def main():
global_config = config['Global'] global_config = config['Global']
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册