提交 c3930dc5 编写于 作者: L LielinJiang

update log level set

上级 0033bc54
...@@ -44,7 +44,7 @@ def setup_logger(output=None, name="hapi", log_level=logging.INFO): ...@@ -44,7 +44,7 @@ def setup_logger(output=None, name="hapi", log_level=logging.INFO):
local_rank = ParallelEnv().local_rank local_rank = ParallelEnv().local_rank
if local_rank == 0: if local_rank == 0:
ch = logging.StreamHandler(stream=sys.stdout) ch = logging.StreamHandler(stream=sys.stdout)
ch.setLevel(logging.DEBUG) ch.setLevel(log_level)
ch.setFormatter(logging.Formatter(format_str)) ch.setFormatter(logging.Formatter(format_str))
logger.addHandler(ch) logger.addHandler(ch)
...@@ -62,7 +62,7 @@ def setup_logger(output=None, name="hapi", log_level=logging.INFO): ...@@ -62,7 +62,7 @@ def setup_logger(output=None, name="hapi", log_level=logging.INFO):
os.makedirs(os.path.dirname(filename)) os.makedirs(os.path.dirname(filename))
fh = logging.StreamHandler(filename) fh = logging.StreamHandler(filename)
fh.setLevel(logging.DEBUG) fh.setLevel(log_level)
fh.setFormatter(logging.Formatter(format_str)) fh.setFormatter(logging.Formatter(format_str))
logger.addHandler(fh) logger.addHandler(fh)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册