`logging.ERROR` on `RANK` not in (0, 1) (#8284)
Improve DDP console response in https://github.com/ultralytics/yolov5/issues/8283pull/8327/head
parent
1156a323f8
commit
a2a1ed201d
utils
|
@ -85,7 +85,7 @@ def set_logging(name=None, verbose=VERBOSE):
|
|||
for h in logging.root.handlers:
|
||||
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
|
||||
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.setLevel(level)
|
||||
handler = logging.StreamHandler()
|
||||
|
|
Loading…
Reference in New Issue