|
|
@@ -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) |