Update prune interface (#3336)

tools/infer.py

@@ -106,12 +106,12 @@ def prune_feed_vars(feeded_var_names, target_vars, prog):
     """
     exist_var_names = []
     prog = prog.clone()
-    prog = prog._prune(targets=target_vars)
+    prog = prog._prune(feeded_var_names, targets=target_vars)
     global_block = prog.global_block()
     for name in feeded_var_names:
         try:
             v = global_block.var(name)
-            exist_var_names.append(v.name)
+            exist_var_names.append(v.name.encode('utf-8'))
         except Exception:
             logger.info('save_inference_model pruned unused feed '
                         'variables {}'.format(name))
@@ -127,8 +127,9 @@ def save_infer_model(FLAGS, exe, feed_vars, test_fetches, infer_prog):
     feeded_var_names = prune_feed_vars(feeded_var_names, target_vars,
                                        infer_prog)
     logger.info("Save inference model to {}, input: {}, output: "
-                "{}...".format(save_dir, feeded_var_names,
-                               [var.name for var in target_vars]))
+                "{}...".format(save_dir, feeded_var_names, [
+                    var.name.encode('utf-8') for var in target_vars
+                ]))
     fluid.io.save_inference_model(
         save_dir,
         feeded_var_names=feeded_var_names,