mirrors
/
mmcv
mirror of https://github.com/open-mmlab/mmcv.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[Feature] Add file mode for logger (#808) 

* add file mode for logger

* add more docs and comments
pull/883/head
Rui Xu 2021-02-02 22:49:24 +08:00 committed by GitHub
parent 93980b4b91
commit cf2f14d06c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mmcv/utils/logging.py
View File

@ -5,7 +5,7 @@ import torch.distributed as dist
logger_initialized = {} logger_initialized = {}
def get_logger(name, log_file=None, log_level=logging.INFO): def get_logger(name, log_file=None, log_level=logging.INFO, file_mode='w'):
"""Initialize and get a logger by name. """Initialize and get a logger by name.
If the logger has not been initialized, this method will initialize the If the logger has not been initialized, this method will initialize the
@ -21,6 +21,8 @@ def get_logger(name, log_file=None, log_level=logging.INFO):
log_level (int): The logger level. Note that only the process of log_level (int): The logger level. Note that only the process of
rank 0 is affected, and other processes will set the level to rank 0 is affected, and other processes will set the level to
"Error" thus be silent most of the time. "Error" thus be silent most of the time.
file_mode (str): The file mode used in opening log file.
Defaults to 'w'.
Returns: Returns:
logging.Logger: The expected logger. logging.Logger: The expected logger.
@ -45,7 +47,10 @@ def get_logger(name, log_file=None, log_level=logging.INFO):
# only rank 0 will add a FileHandler # only rank 0 will add a FileHandler
if rank == 0 and log_file is not None: if rank == 0 and log_file is not None:
file_handler = logging.FileHandler(log_file, 'w') # Here, the default behaviour of the official logger is 'a'. Thus, we
# provide an interface to change the file mode to the default
# behaviour.
file_handler = logging.FileHandler(log_file, file_mode)
handlers.append(file_handler) handlers.append(file_handler)
formatter = logging.Formatter( formatter = logging.Formatter(