From 86b17393ebdbb475666b9b652472431b06f0f0c6 Mon Sep 17 00:00:00 2001 From: wangguanzhong Date: Tue, 8 Oct 2019 14:27:55 +0800 Subject: [PATCH] fix prune interface (#3472) --- PaddleCV/PaddleDetection/tools/infer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/PaddleCV/PaddleDetection/tools/infer.py b/PaddleCV/PaddleDetection/tools/infer.py index d33d1e3e..64049e3f 100644 --- a/PaddleCV/PaddleDetection/tools/infer.py +++ b/PaddleCV/PaddleDetection/tools/infer.py @@ -106,7 +106,7 @@ def prune_feed_vars(feeded_var_names, target_vars, prog): """ exist_var_names = [] prog = prog.clone() - prog = prog._prune(feeded_var_names, targets=target_vars) + prog = prog._prune(targets=target_vars) global_block = prog.global_block() for name in feeded_var_names: try: -- GitLab