Improve DDP console response in https://github.com/ultralytics/yolov5/issues/8283modifyDataloader
@@ -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() |