Kaggle `LOGGER` fix (#6041)
Browse files- utils/general.py +2 -0
utils/general.py
CHANGED
@@ -46,6 +46,8 @@ os.environ['NUMEXPR_MAX_THREADS'] = str(NUM_THREADS) # NumExpr max threads
|
|
46 |
|
47 |
def set_logging(name=None, verbose=True):
|
48 |
# Sets level and returns logger
|
|
|
|
|
49 |
rank = int(os.getenv('RANK', -1)) # rank in world for Multi-GPU trainings
|
50 |
logging.basicConfig(format="%(message)s", level=logging.INFO if (verbose and rank in (-1, 0)) else logging.WARNING)
|
51 |
return logging.getLogger(name)
|
|
|
46 |
|
47 |
def set_logging(name=None, verbose=True):
|
48 |
# Sets level and returns logger
|
49 |
+
for h in logging.root.handlers[:]:
|
50 |
+
logging.root.removeHandler(h) # remove all handlers associated with the root logger object
|
51 |
rank = int(os.getenv('RANK', -1)) # rank in world for Multi-GPU trainings
|
52 |
logging.basicConfig(format="%(message)s", level=logging.INFO if (verbose and rank in (-1, 0)) else logging.WARNING)
|
53 |
return logging.getLogger(name)
|