Browse Source

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

Improve DDP console response in https://github.com/ultralytics/yolov5/issues/8283
modifyDataloader
Glenn Jocher GitHub 2 years ago
parent
commit
a2a1ed201d
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      utils/general.py

+ 1
- 1
utils/general.py View File

@@ -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…
Cancel
Save