使用TensorFlow时,可能会出现一些Warning等等的日志信息。
这些warning通常是版本问题和旧方法在新版本里抛弃,没有什么大的用处,但是过多的warning日志信息会影响到正常的debug。那应该如何屏蔽TensorFlow的日志信息呢?
出现一些Warning等等的日志信息是因为TensorFlow的默认日志等级为0。TensorFlow的日志等级由环境变量TF_CPP_MIN_LOG_LEVEL决定,将日志等级设置为1可以屏蔽INFO级别的日志,设置为2可以屏蔽WARNING级别的日志,设置为3可以屏蔽ERROR级别的日志,具体方法如下:
import os
import tensorflow as tf
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
网上大多数博客都是这样写的,但是我发现在我的代码中并不适用,我的TensorFlow版本是1.15.0所以我猜测是版本不适用的问题。
那么就可以使用如下方法将日志等级设置为DEBUG、INFO、WARNING、ERROR或者FATAL:
tf.logging.set_verbosity(tf.logging.ERROR)
成功解决:
Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 FMA的问题:
参考:https://blog.csdn.net/qq_41185868/article/details/79127838
只需要将两者结合起来:
import os
import tensorflow as tf
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
tf.logging.set_verbosity(tf.logging.ERROR)
完美解决。