Python中Warning Control的一些理解

写在前面

最近在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(