【Python深度学习】——使用Logging记录模型训练过程

本文主要是介绍【Python深度学习】——使用Logging记录模型训练过程，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

1. 导入logging库, getLogger()创建日志记录器

import logging
logger = logging.getLogger()

在getLogger()函数中,可以设置日志记录器的名称, 例如:

logger = logging.getLogger(__name__)
## 或者
logger = logging.getLogger('my_logger')

2. logger.setLevel()设置log级别

logger.setLevel(level=logging.DEBUG)

级别	示例	级别数值
DEBUG	logging.debug	10
INFO	logging.info	20
WARNING	logging.warning	30
ERROR	logging.error	40
CRITICAL	logging.critical	50

3. FileHandler()设置日志文件路径

file_handler = logging.FileHandler('my_log.log')

4. StreamHandler()将日志输出到控制台

stream_handler = logging.StreamHandler()
logger.addHandler(stream_handler)

5. 其他性质

• propagate: 是否将日志消息传递给上级日志记录器

logger.propagate = False

6. 代码示例

在Geo Transformer代码中, 对logging库中要使用的一些功能, 封装成了一个Logger类, 详情如下:

import loggingimport coloredlogsdef create_logger(log_file=None):logger = logging.getLogger()logger.handlers.clear()logger.setLevel(level=logging.DEBUG)logger.propagate = Falseformat_str = '[%(asctime)s] [%(levelname).4s] %(message)s'stream_handler = logging.StreamHandler()colored_formatter = coloredlogs.ColoredFormatter(format_str)stream_handler.setFormatter(colored_formatter)logger.addHandler(stream_handler)if log_file is not None:file_handler = logging.FileHandler(log_file)formatter = logging.Formatter(format_str, datefmt='%Y-%m-%d %H:%M:%S')file_handler.setFormatter(formatter)logger.addHandler(file_handler)return loggerclass Logger:def __init__(self, log_file=None, local_rank=-1):if local_rank == 0 or local_rank == -1:self.logger = create_logger(log_file=log_file)else:self.logger = Nonedef debug(self, message):if self.logger is not None:self.logger.debug(message)def info(self, message):if self.logger is not None:self.logger.info(message)def warning(self, message):if self.logger is not None:self.logger.warning(message)def error(self, message):if self.logger is not None:self.logger.error(message)def critical(self, message):if self.logger is not None:self.logger.critical(message)

这篇关于【Python深度学习】——使用Logging记录模型训练过程的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---