【终极解决方案】Tensorflow中忽略提示信息的两种方式

王摇摆

于 2023-05-31 09:16:24 发布

阅读量907

点赞数 1

分类专栏： Tensorflow 文章标签： tensorflow 人工智能 python

本文链接：https://blog.csdn.net/weixin_44943389/article/details/130961791

版权

2 篇文章 0 订阅

订阅专栏

要禁止输出TensorFlow的警告信息，您可以使用Python的日志记录模块来控制日志级别。在导入TensorFlow之前，您可以添加以下代码来设置日志级别为"ERROR"，这样只会显示错误级别的日志，而不会显示警告信息：

import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'  # 设置TensorFlow日志级别为"ERROR"
import tensorflow as tf

这将使TensorFlow仅显示错误级别的日志，而不会显示警告信息。请注意，这样做可能会导致您错过一些有用的警告信息，因此请在调试和开发期间小心使用。

另外，如果您使用的是旧版本的TensorFlow（例如TensorFlow 1.x），您可以使用以下代码来禁止TensorFlow输出警告信息：

import warnings
warnings.filterwarnings("ignore")  # 忽略警告信息
import tensorflow as tf

这将通过过滤警告信息来禁止TensorFlow的警告输出。请注意，这种方法不适用于TensorFlow 2.x版本。

关注