fix the assignment of a var

5 years ago · af8740330b
parent cd56032479
commit af8740330b
1 changed files with 4 additions and 3 deletions
--- 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