tensorflow日志信息如何屏蔽?

昨天跑模型,发现测试的时候出了一大堆的warning,于是查了一下如何屏蔽日志信息:

os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
tf.logging.set_verbosity(tf.logging.ERROR)

出现一些Warning等等的日志信息是因为TensorFlow的默认日志等级为0。

TensorFlow的日志等级由环境变量TF_CPP_MIN_LOG_LEVEL决定,将日志等级设置为1可以屏蔽INFO级别的日志,设置为2可以屏蔽WARNING级别的日志,设置为3可以屏蔽ERROR级别的日志。(代码第一行)

网上大多数博客都是只写了这一行,但是我发现在我的代码中并不适用,我的TensorFlow版本是1.15.0所以我猜测是版本不适用的问题。
那么就可以使用如下方法将日志等级设置为DEBUG、INFO、WARNING、ERROR或者FATAL:

 

原文链接:https://blog.csdn.net/joey_ro/article/details/117196932

posted @ 2021-10-13 15:25  Tomorrow1126  阅读(229)  评论(0编辑  收藏  举报