mirror of
https://github.com/botastic/SoftGroup.git
synced 2025-10-16 11:45:42 +00:00
22 lines
720 B
Python
22 lines
720 B
Python
from .dist import get_dist_info
|
|
import logging
|
|
|
|
|
|
def get_root_logger(log_file=None, log_level=logging.INFO):
|
|
logger = logging.getLogger('pc-seg')
|
|
# if the logger has been initialized, just return it
|
|
if logger.hasHandlers():
|
|
return logger
|
|
|
|
logging.basicConfig(format='%(asctime)s - %(levelname)s - %(message)s', level=log_level)
|
|
rank, _ = get_dist_info()
|
|
if rank != 0:
|
|
logger.setLevel('ERROR')
|
|
elif log_file is not None:
|
|
file_handler = logging.FileHandler(log_file, 'w')
|
|
file_handler.setFormatter(logging.Formatter('%(asctime)s - %(levelname)s - %(message)s'))
|
|
file_handler.setLevel(log_level)
|
|
logger.addHandler(file_handler)
|
|
|
|
return logger
|