|
|
|
|
|
|
|
|
|
import os.path as osp |
|
import time |
|
import sys |
|
import logging |
|
|
|
import torch.distributed as dist |
|
|
|
|
|
def setup_logger(logpth): |
|
logfile = 'BiSeNet-{}.log'.format(time.strftime('%Y-%m-%d-%H-%M-%S')) |
|
logfile = osp.join(logpth, logfile) |
|
FORMAT = '%(levelname)s %(filename)s(%(lineno)d): %(message)s' |
|
log_level = logging.INFO |
|
if dist.is_initialized() and not dist.get_rank()==0: |
|
log_level = logging.ERROR |
|
logging.basicConfig(level=log_level, format=FORMAT, filename=logfile) |
|
logging.root.addHandler(logging.StreamHandler()) |
|
|
|
|
|
|
