TensorFlow中屏蔽warning的方法

 TensorFlow的日志级别分为如下三种:         python

  1.  TF_CPP_MIN_LOG_LEVEL = 1         //默认设置,为显示全部信息                
  2. TF_CPP_MIN_LOG_LEVEL = 2         //只显示error和warining信息                
  3. TF_CPP_MIN_LOG_LEVEL = 3         //只显示error信息

因此,当TensorFlow出现警告信息,又不想让警告信息显示时,可进行以下设置:spa

 

python环境下经过在python文件中添加以下两行代码,设置TensorFlow日志输出级别。日志

1 import os
2 os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2"
相关文章
相关标签/搜索