`logging.ERROR` on `RANK` not in (0, 1) (#8284)
Browse filesImprove DDP console response in https://github.com/ultralytics/yolov5/issues/8283
- 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.
|
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()
|