Merge pull request #7830 from wanghaoshuang/fix_infer_io

Fix get_inference_program
emailweixu-patch-1
whs 7 years ago committed by GitHub
commit f73e5e195e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -191,8 +191,8 @@ def get_inference_program(target_vars, main_program=None):
vars = []
for var in target_vars:
if isinstance(var, Evaluator):
vars.append(var.states)
vars.append(var.metrics)
vars.extend(var.states)
vars.extend(var.metrics)
else:
vars.append(var)
pruned_program = main_program.prune(targets=vars)

Loading…
Cancel
Save