fix prune interface (#3472)

3ed4fd64 · wangguanzhong · GitHub · da4bb135 · 3ed4fd64
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

tools/infer.py tools/infer.py +1 -1

未找到文件。
--- a/tools/infer.py
+++ b/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: