lesson-3

haolangchen · May 5, 2020 · 23db136 · 23db136
1 parent 1e6c5d9
commit 23db136
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/Pytorch-Seg/lesson-3/logger.py b/Pytorch-Seg/lesson-3/logger.py
@@ -0,0 +1,19 @@
+import logging
+
+def get_logger(LEVEL, log_file = None):
+    head = '[%(asctime)-15s] [%(levelname)s] %(message)s'
+    if LEVEL == 'info':
+        logging.basicConfig(level=logging.INFO, format=head)
+    elif LEVEL == 'debug':
+        logging.basicConfig(level=logging.DEBUG, format=head)
+    logger = logging.getLogger()
+    if log_file != None:
+        fh = logging.FileHandler(log_file)
+        logger.addHandler(fh)
+    return logger
+
+logger = get_logger('info')
+
+logger.info('Jack Cui')
+logger.info('https://cuijiahua.com')
+logger.info('https://mp.weixin.qq.com/s/OCWwRVDFNslIuKyiCVUoTA')