Simplify set_logging() indexing (#6042)
This commit is contained in:
1 parent
0db9d5b6a2
commit
b8a4babd60
1 file changed
+1
-1
+1
-1
@@ -46,7 +46,7 @@ os.environ['NUMEXPR_MAX_THREADS'] = str(NUM_THREADS) # NumExpr max threads
|
||||
|
||||
def set_logging(name=None, verbose=True):
|
||||
# Sets level and returns logger
|
||||
for h in logging.root.handlers[:]:
|
||||
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
|
||||
logging.basicConfig(format="%(message)s", level=logging.INFO if (verbose and rank in (-1, 0)) else logging.WARNING)
|
||||
|
||||
Reference in new issue
Block a user