TensorFlow中屏蔽warning的方法

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

  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出现警告信息,又不想让警告信息显示时,可进行如下设置:

 

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

1 import os
2 os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2"

 

posted @ 2019-10-10 11:45  Weisswire  阅读(1629)  评论(0编辑  收藏  举报