导入PyTorch模型加载函数时出现大量调试信息泛滥问题

问题原因 #

这些DEBUG日志来自两个依赖库：

• git（gitpython库）在执行版本检查时输出调试信息
• numba在JIT编译过程中输出内部SSA分析的调试日志

本质是Python的logging模块被设置为DEBUG级别，或者这些库自身默认开启了DEBUG日志输出。

解决方案 #

方法1：全局调整日志级别 #

在所有模块导入前，将全局日志级别设置为INFO或更高（如WARNING），直接屏蔽所有DEBUG级别的日志输出：

import logging
# 设置全局日志级别为INFO，仅显示INFO及以上级别的日志
logging.basicConfig(level=logging.INFO)

# 之后再导入你的模块
from data.src.dataLoaders import GrooveDataSet_Density
from model import GrooVAEDensity1D
from helpers.Control.density_1D_modelLoader import load_density_1d_model

方法2：针对性禁用特定库的日志 #

如果需要保留其他库的DEBUG日志，仅屏蔽numba和git的日志：

import logging
# 禁用numba的DEBUG日志
logging.getLogger('numba').setLevel(logging.WARNING)
# 禁用git相关的DEBUG日志
logging.getLogger('git').setLevel(logging.WARNING)

# 导入模块
from data.src.dataLoaders import GrooveDataSet_Density
from model import GrooVAEDensity1D
from helpers.Control.density_1D_modelLoader import load_density_1d_model

方法3：检查代码中的日志配置 #

确认你的项目代码或依赖库中有没有手动设置logging.basicConfig(level=logging.DEBUG)，如果有，将其修改为更高的级别即可。

验证效果 #

添加上述日志配置后，再运行代码就不会出现大量的DEBUG:git.cmd...和DEBUG:numba.core.ssa...这类输出了。

内容的提问来源于stack exchange，提问作者JLenz