From af8740330b746e6225b8ef7f18f79d0aeb5bd2fc Mon Sep 17 00:00:00 2001 From: "shaohua.zhang" Date: Thu, 23 Jul 2020 13:59:58 +0800 Subject: [PATCH] fix the assignment of a var --- tools/program.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tools/program.py b/tools/program.py index c39e0878..4ebc1167 100755 --- a/tools/program.py +++ b/tools/program.py @@ -394,8 +394,9 @@ def preprocess(): startup_program = fluid.Program() train_program = fluid.Program() - contain_det = False if alg in ['EAST', 'DB']: - contain_det = True + train_alg_type = 'det' + else: + train_alg_type = 'rec' - return startup_program, train_program, place, config, contain_det + return startup_program, train_program, place, config, train_alg_type -- GitLab