如何在 Python 中忽略烦人的警告?

如何在 Python 中忽略烦人的警告?

    • 代码讲解
      • [**1. 导入需要用到的工具**](#1. 导入需要用到的工具)
      • [**2. 定义一个函数:ignore_warnings**](#2. 定义一个函数:ignore_warnings)
      • [**3. 忽略特定的警告**](#3. 忽略特定的警告)
      • [**4. 调整日志显示级别**](#4. 调整日志显示级别)
      • [**5. 设置 Transformers 库的日志等级**](#5. 设置 Transformers 库的日志等级)
    • 什么时候用这段代码?
    • 总结

当你用 Python 写代码时,有时候会看到一些"警告"信息。这些信息不会让代码出错,但会让输出看起来很乱。如果你觉得这些警告很烦,可以用下面的代码让它们"闭嘴"。今天我们就来学学这段代码是怎么做到的!


代码讲解

1. 导入需要用到的工具

python 复制代码
import warnings
import logging
from transformers import logging as hf_logging
  • warnings:这是 Python 内置的工具,用来管理"警告"。
  • logging:这是 Python 的日志工具,日志就是程序运行时输出的提示信息。
  • hf_logging :这是 transformers 库的日志工具,我们也可以用它调整日志的显示。

2. 定义一个函数:ignore_warnings

python 复制代码
def ignore_warnings():

函数就像一个"工具箱",当你需要它的功能时,只要调用它就可以。


3. 忽略特定的警告

python 复制代码
warnings.filterwarnings("ignore", message="Some weights of the model checkpoint")
warnings.filterwarnings("ignore", message="Could not find image processor class")
warnings.filterwarnings("ignore", message="The `max_size` parameter is deprecated")
  • warnings.filterwarnings("ignore", ...):这是说"我要忽略某些警告"。
  • message:这里写的是警告的内容,只要警告里包含这段文字,就会被"静音"。

就像我们告诉程序:"看到这些'内容'的警告,别烦我!"


4. 调整日志显示级别

python 复制代码
logging.basicConfig(level=logging.ERROR)
  • logging.basicConfig:设置日志的"声音大小"。
  • level=logging.ERROR:告诉程序"只显示出错的信息,别展示其他没用的提示"。

5. 设置 Transformers 库的日志等级

python 复制代码
hf_logging.set_verbosity_error()
  • hf_logging.set_verbosity_error() :这是专门用来调节 transformers 库的日志,告诉它"只显示出错信息,其他的不要说话"。

什么时候用这段代码?

  • 用深度学习工具 :比如 transformers,它会输出很多没用的警告,影响阅读。
  • 调试代码时:有时候只想看重要的信息,忽略那些不必要的提示。

总结

这段代码就是一把"静音遥控器",帮我们把那些烦人的警告和不重要的信息屏蔽掉,让代码的输出更加干净整洁!

完整代码

python 复制代码
import warnings
import logging
from transformers import logging as hf_logging

def ignore_warnings():
    # 忽略特定的警告
    warnings.filterwarnings("ignore", message="Some weights of the model checkpoint")
    warnings.filterwarnings("ignore", message="Could not find image processor class")
    warnings.filterwarnings("ignore", message="The `max_size` parameter is deprecated")

    # 调整日志等级,只显示错误信息
    logging.basicConfig(level=logging.ERROR)
    hf_logging.set_verbosity_error()

快把这段代码加到你的项目里试试吧!它会让你的输出世界清静很多哦 😊。

相关推荐
lucky_lyovo3 小时前
自然语言处理NLP---预训练模型与 BERT
人工智能·自然语言处理·bert
fantasy_arch3 小时前
pytorch例子计算两张图相似度
人工智能·pytorch·python
七七&5563 小时前
2024年08月13日 Go生态洞察:Go 1.23 发布与全面深度解读
开发语言·网络·golang
java坤坤3 小时前
GoLand 项目从 0 到 1:第八天 ——GORM 命名策略陷阱与 Go 项目启动慢问题攻坚
开发语言·后端·golang
元清加油3 小时前
【Golang】:函数和包
服务器·开发语言·网络·后端·网络协议·golang
健康平安的活着4 小时前
java之 junit4单元测试Mockito的使用
java·开发语言·单元测试
AndrewHZ4 小时前
【3D重建技术】如何基于遥感图像和DEM等数据进行城市级高精度三维重建?
图像处理·人工智能·深度学习·3d·dem·遥感图像·3d重建
飞哥数智坊4 小时前
Coze实战第18讲:Coze+计划任务,我终于实现了企微资讯简报的定时推送
人工智能·coze·trae
WBluuue4 小时前
数学建模:智能优化算法
python·机器学习·数学建模·爬山算法·启发式算法·聚类·模拟退火算法
Code_流苏5 小时前
AI热点周报(8.10~8.16):AI界“冰火两重天“,GPT-5陷入热议,DeepSeek R2模型训练受阻?
人工智能·gpt·gpt5·deepseek r2·ai热点·本周周报