diff --git a/tools/train.py b/tools/train.py
index bee9214d99c14217abb855e6c178d3e6ffc26929..149490123599fd194c662c90937736444c00f150 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()