mirror of https://github.com/kritiksoman/GIMP-ML
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
24 lines
598 B
Python
24 lines
598 B
Python
#!/usr/bin/python
|
|
# -*- encoding: utf-8 -*-
|
|
|
|
|
|
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())
|
|
|
|
|