From ed454d1423db0b5229185f9df67db0015d4004e7 Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Wed, 1 Jul 2020 13:09:44 +0800 Subject: [PATCH] print to logger.info --- tools/train.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/tools/train.py b/tools/train.py index 20f63c25..15d6ebb2 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__': -- GitLab