Update program.py

PaddlePaddle · May 15, 2020 · f5fb36e · f5fb36e
1 parent 28ab6a3
commit f5fb36e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tools/program.py b/tools/program.py
@@ -191,8 +191,8 @@ def build_export(config, main_prog, startup_prog):
  func_infor = config['Architecture']['function']
  model = create_module(func_infor)(params=config)
  image, outputs = model(mode='export')
- fetches_var = sorted([outputs[name] for name in outputs])
- fetches_var_name = [name for name in fetches_var]
+ fetches_var_name = sorted([name for name in outputs])
+ fetches_var = [outputs[name] for name in fetches_var_name]
  feeded_var_names = [image.name]
  target_vars = fetches_var
  return feeded_var_names, target_vars, fetches_var_name