diff --git a/tools/train.py b/tools/train.py
index bee9214d..14949012 100755
--- a/tools/train.py
+++ b/tools/train.py
@@ -117,6 +117,6 @@ def test_reader():
 
 
 if __name__ == '__main__':
-    startup_program, train_program, place, config, isContain_det = program.preprocess()
+    startup_program, train_program, place, config, contain_det = program.preprocess()
     main()
 #     test_reader()