diff --git a/tools/train.py b/tools/train.py
index 20f63c25ac7a62003076f18ce7dbb2638060323e..15d6ebb2138ce19a2f65c7d1fabd56d86b7645be 100755
--- a/tools/train.py
+++ b/tools/train.py
@@ -122,9 +122,8 @@ def test_reader():
                 starttime = time.time()
                 print("reader:", count, len(data), batch_time)
     except Exception as e:
-        print(e)
-    print("finish reader:", count)
-    print("success")
+        logger.info(e)
+    logger.info("finish reader: {}, Success!".format(count))
 
 
 if __name__ == '__main__':