TensorFlow的日志级别分为以下三种:
- TF_CPP_MIN_LOG_LEVEL = 1 //默认设置,为显示所有信息
- TF_CPP_MIN_LOG_LEVEL = 2 //只显示error和warining信息
- TF_CPP_MIN_LOG_LEVEL = 3 //只显示error信息
所以,当TensorFlow出现警告信息,又不想让警告信息显示时,可进行如下设置:
python环境下通过在python文件中添加如下两行代码,设置TensorFlow日志输出级别。
1 import os 2 os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2"
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:TensorFlow中屏蔽warning的方法 - Python技术站