写在前面
最近在pytorch运行网络模型时,因为版本的问题,有大量的warning输出,这对于根据输出查看训练状态来说是很麻烦的,所以有了忽略warning输出的需求.
Python中的warning Control
基本概念
warning类型分类
Python中有多种warning类型,其中Warning
为其他所有warning类型的基类.详情如下:
UserWarning
: 用户警告DeprecationWarning
:弃用警告SyntaxWarning
: 语法警告RuntimeWarning
: 运行时警告FutureWarning
: 未来可能发生的警告PendingDeprecationWarning
: 未来可能被弃用的警告ImportWarning
: 导入警告UnicodeWarning
: 与Unicode编码相关的警告BytesWarning
: 与字节或者字节数组相关的警告ResourceWarning
: 与资源使用相关的警告
详情可查看此处
warning filter
警告过滤器可以理解为是一个表中的一个条目,格式为(action, message, category, module, lineno)
,其中,action
表示匹配到的警告采用什么样的频率输出, message
表示警告输出信息,此处是一个正则匹配项,category
表示匹配的warning类型, module
表示警告发出所在的模块名,也是正则匹配项, lineno
表示警告发出的行号,只有(message, category, module, lineno)
都匹配到发出的警告时,相应的action
才会进行.此处我们需要简单的理解一个警告发出的过程
- 首先调用
warnings.warn(message, category)
表示此处有个警告需发出 - 然后根据给定的警告类型
category
,在warning filter构成的列表中寻找匹配项,找到匹配的filter, 然后对该类型的警告执行action
操作,如果没有匹配的filter则对该类型警告执行默认操作.
因此我们可以看出warning filter的作用是用于控制警告的输出动作.
cation分类
action
表示对于将发出的警告需执行的动作,包含着输出的频率
default
: 对于匹配的警告,打印在每个位置第一次出现时输出(模块+行号),之后在同一位置出现时不输出.error
: 将匹配的警告变为异常ignore
: 忽略警告,不进行输出always
: 总是打印输出匹配的警告module
: 对于匹配到的警告,在一个模块中第一次出现时才输出(与行号无关),之后不管在模块的哪个位置出现都不输出.once
: 对于匹配到的警告只打印输出一次,而不管在哪个位置.
warn代码原理剖析
警告发出的关键方法是warnings.warn()
,在warnings.py文件中,因此可以深入去理解该部分代码,部分进行了省略.
def warn(message, category=None, stacklevel=1, source=None):
# 1.类型检查
# 2.设置行号,设置模块名, 设置文件名
registry = globals.setdefault(