fix the assignment of a var

af874033 · shaohua.zhang · GitHub · cd560324 · af874033
显示空白变更内容
内联并排

Showing with 4 addition and 3 deletion

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

未找到文件。
--- 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