glenn-jocher commited on
Commit
a2a1ed2
·
unverified ·
1 Parent(s): 1156a32

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

Browse files

Improve DDP console response in https://github.com/ultralytics/yolov5/issues/8283

Files changed (1) hide show
  1. utils/general.py +1 -1
utils/general.py CHANGED
@@ -85,7 +85,7 @@ def set_logging(name=None, verbose=VERBOSE):
85
  for h in logging.root.handlers:
86
  logging.root.removeHandler(h) # remove all handlers associated with the root logger object
87
  rank = int(os.getenv('RANK', -1)) # rank in world for Multi-GPU trainings
88
- level = logging.INFO if verbose and rank in {-1, 0} else logging.WARNING
89
  log = logging.getLogger(name)
90
  log.setLevel(level)
91
  handler = logging.StreamHandler()
 
85
  for h in logging.root.handlers:
86
  logging.root.removeHandler(h) # remove all handlers associated with the root logger object
87
  rank = int(os.getenv('RANK', -1)) # rank in world for Multi-GPU trainings
88
+ level = logging.INFO if verbose and rank in {-1, 0} else logging.ERROR
89
  log = logging.getLogger(name)
90
  log.setLevel(level)
91
  handler = logging.StreamHandler()