fix the assignment of a var

af874033 · shaohua.zhang · GitHub · cd560324 · af874033
Unverified Commit af874033 authored Jul 23, 2020 by shaohua.zhang Committed by GitHub Jul 23, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 3 deletions

tools/program.py tools/program.py +4 -3

No files found.
--- 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