diff --git a/PaddleCV/PaddleDetection/tools/infer.py b/PaddleCV/PaddleDetection/tools/infer.py index d33d1e3ebecaa342643817c1302220f202d93885..64049e3fc2b513d6e1e8946edc4bb33518ea8f40 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: