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

`logging.ERROR` on `RANK` not in (0, 1) (#8284)

Improve DDP console response in https://github.com/ultralytics/yolov5/issues/8283
上级 1156a323
...@@ -85,7 +85,7 @@ def set_logging(name=None, verbose=VERBOSE): ...@@ -85,7 +85,7 @@ def set_logging(name=None, verbose=VERBOSE):
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
level = logging.INFO if verbose and rank in {-1, 0} else logging.WARNING level = logging.INFO if verbose and rank in {-1, 0} else logging.ERROR
log = logging.getLogger(name) log = logging.getLogger(name)
log.setLevel(level) log.setLevel(level)
handler = logging.StreamHandler() handler = logging.StreamHandler()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论