提交 3e9f8056 编写于 作者: L LielinJiang

fix incompatibility and remove basicConfig

上级 f4f461bd
...@@ -15,12 +15,10 @@ ...@@ -15,12 +15,10 @@
import os import os
import sys import sys
import logging import logging
import functools
from paddle.fluid.dygraph.parallel import ParallelEnv from paddle.fluid.dygraph.parallel import ParallelEnv
@functools.lru_cache()
def setup_logger(output=None, name="hapi", log_level=logging.INFO): def setup_logger(output=None, name="hapi", log_level=logging.INFO):
""" """
Initialize logger of hapi and set its verbosity level to "INFO". Initialize logger of hapi and set its verbosity level to "INFO".
...@@ -36,9 +34,7 @@ def setup_logger(output=None, name="hapi", log_level=logging.INFO): ...@@ -36,9 +34,7 @@ def setup_logger(output=None, name="hapi", log_level=logging.INFO):
""" """
logger = logging.getLogger(name) logger = logging.getLogger(name)
logger.propagate = False logger.propagate = False
logger.setLevel(log_level)
format_str = '%(asctime)s - %(name)s - %(levelname)s - %(message)s'
logging.basicConfig(format=format_str, level=log_level)
# stdout logging: only local rank==0 # stdout logging: only local rank==0
local_rank = ParallelEnv().local_rank local_rank = ParallelEnv().local_rank
...@@ -46,6 +42,7 @@ def setup_logger(output=None, name="hapi", log_level=logging.INFO): ...@@ -46,6 +42,7 @@ def setup_logger(output=None, name="hapi", log_level=logging.INFO):
ch = logging.StreamHandler(stream=sys.stdout) ch = logging.StreamHandler(stream=sys.stdout)
ch.setLevel(log_level) ch.setLevel(log_level)
format_str = '%(asctime)s - %(name)s - %(levelname)s - %(message)s'
ch.setFormatter(logging.Formatter(format_str)) ch.setFormatter(logging.Formatter(format_str))
logger.addHandler(ch) logger.addHandler(ch)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册