Unverified 提交 b8a4babd authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

Simplify `set_logging()` indexing (#6042)

上级 0db9d5b6
...@@ -46,7 +46,7 @@ os.environ['NUMEXPR_MAX_THREADS'] = str(NUM_THREADS) # NumExpr max threads ...@@ -46,7 +46,7 @@ os.environ['NUMEXPR_MAX_THREADS'] = str(NUM_THREADS) # NumExpr max threads
def set_logging(name=None, verbose=True): def set_logging(name=None, verbose=True):
# Sets level and returns logger # Sets level and returns logger
for h in logging.root.handlers[:]: for h in logging.root.handlers:
logging.root.removeHandler(h) # remove all handlers associated with the root logger object logging.root.removeHandler(h) # remove all handlers associated with the root logger object
rank = int(os.getenv('RANK', -1)) # rank in world for Multi-GPU trainings rank = int(os.getenv('RANK', -1)) # rank in world for Multi-GPU trainings
logging.basicConfig(format="%(message)s", level=logging.INFO if (verbose and rank in (-1, 0)) else logging.WARNING) logging.basicConfig(format="%(message)s", level=logging.INFO if (verbose and rank in (-1, 0)) else logging.WARNING)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论