print to logger.info

ed454d14 · LDOUBLEV · 3892a8ca · ed454d14
Commit ed454d14 authored Jul 01, 2020 by LDOUBLEV
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 3 deletions

tools/train.py tools/train.py +2 -3

No files found.
--- 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)
+        logger.info(e)
-    print("finish reader:", count)
+    logger.info("finish reader: {}, Success!".format(count))
-    print("success")
 if __name__ == '__main__':