diff --git a/tools/program.py b/tools/program.py
index c39e0878e84ec4e6897e5f15b817e3e7db1cc3e2..4ebc11670702ba627c89b060692c9827e6e163fd 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