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