From c3930dc53666f1c8d0ba6345222befcb9c53dc97 Mon Sep 17 00:00:00 2001
From: LielinJiang <jianglielin@baidu.com>
Date: Mon, 27 Apr 2020 05:35:25 +0000
Subject: [PATCH] update log level set

---
 hapi/logger.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/hapi/logger.py b/hapi/logger.py
index 229bd13..8297f3e 100644
--- a/hapi/logger.py
+++ b/hapi/logger.py
@@ -44,7 +44,7 @@ def setup_logger(output=None, name="hapi", log_level=logging.INFO):
     local_rank = ParallelEnv().local_rank
     if local_rank == 0:
         ch = logging.StreamHandler(stream=sys.stdout)
-        ch.setLevel(logging.DEBUG)
+        ch.setLevel(log_level)
 
         ch.setFormatter(logging.Formatter(format_str))
         logger.addHandler(ch)
@@ -62,7 +62,7 @@ def setup_logger(output=None, name="hapi", log_level=logging.INFO):
             os.makedirs(os.path.dirname(filename))
 
         fh = logging.StreamHandler(filename)
-        fh.setLevel(logging.DEBUG)
+        fh.setLevel(log_level)
         fh.setFormatter(logging.Formatter(format_str))
         logger.addHandler(fh)
 
-- 
GitLab