print to logger.info

2020-07-01 13:09:44 +08:00 · 2020-07-01 13:09:44 +08:00 · ed454d1423
parent 3892a8ca02
commit ed454d1423
1 changed files with 2 additions and 3 deletions
--- 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__':