提交 0539966b 编写于 作者: L LielinJiang

fix logger in python2

上级 31766e3a
...@@ -36,13 +36,13 @@ def setup_logger(output=None, name="hapi", log_level=logging.INFO): ...@@ -36,13 +36,13 @@ def setup_logger(output=None, name="hapi", log_level=logging.INFO):
logger.propagate = False logger.propagate = False
logger.setLevel(log_level) logger.setLevel(log_level)
format_str = '%(asctime)s - %(name)s - %(levelname)s - %(message)s'
# stdout logging: only local rank==0 # stdout logging: only local rank==0
local_rank = ParallelEnv().local_rank local_rank = ParallelEnv().local_rank
if local_rank == 0 and not logger.hasHandlers(): if local_rank == 0 and len(logger.handlers) == 0:
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)
...@@ -52,6 +52,7 @@ def setup_logger(output=None, name="hapi", log_level=logging.INFO): ...@@ -52,6 +52,7 @@ def setup_logger(output=None, name="hapi", log_level=logging.INFO):
filename = output filename = output
else: else:
filename = os.path.join(output, "log.txt") filename = os.path.join(output, "log.txt")
if local_rank > 0: if local_rank > 0:
filename = filename + ".rank{}".format(local_rank) filename = filename + ".rank{}".format(local_rank)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册