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