提交 81fd696e 编写于 作者: W wangguanzhong 提交者: GitHub

fix log in python3 (#3349)

上级 78a4d0bf
......@@ -111,7 +111,7 @@ def prune_feed_vars(feeded_var_names, target_vars, prog):
for name in feeded_var_names:
try:
v = global_block.var(name)
exist_var_names.append(v.name.encode('utf-8'))
exist_var_names.append(str(v.name))
except Exception:
logger.info('save_inference_model pruned unused feed '
'variables {}'.format(name))
......@@ -127,9 +127,8 @@ 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.encode('utf-8') for var in target_vars
]))
"{}...".format(save_dir, feeded_var_names,
[str(var.name) for var in target_vars]))
fluid.io.save_inference_model(
save_dir,
feeded_var_names=feeded_var_names,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册