From f5fb36e88c11a41908530b37069dfa154e09602e Mon Sep 17 00:00:00 2001 From: dyning Date: Fri, 15 May 2020 22:25:37 +0800 Subject: [PATCH] Update program.py --- tools/program.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tools/program.py b/tools/program.py index b6318c3b..a114b1cb 100755 --- 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 -- GitLab