print to logger.info

ed454d14 · LDOUBLEV · 3892a8ca · ed454d14
显示空白变更内容
内联并排

Showing with 2 addition and 3 deletion

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

未找到文件。
--- 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__':