tensorflow keras 关闭打印warning

最新推荐文章于 2024-01-01 05:45:00 发布

原创最新推荐文章于 2024-01-01 05:45:00 发布 · 664 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#tensorflow #keras #人工智能

通过设置环境变量TF_CPP_MIN_LOG_LEVEL为3，并使用tf.get_logger().setLevel(ERROR)，可以确保在TensorFlow2.3中屏蔽所有警告信息，只显示错误级别日志。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

用下面代码设置TensorFlow的日志级别为ERROR，以及将环境变量TF_CPP_MIN_LOG_LEVEL设置为'3'，这将禁用TensorFlow的所有警告信息。os.environ部分一定要放在import tensorflow之前，我用的是tensorflow2.3，不加tf.get_logger().setLevel('ERROR')也会不生效。

import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'

import tensorflow as tf
tf.get_logger().setLevel('ERROR')

import os
os.environ["TF_CPP_MIN_LOG_LEVEL"]='1'
# 默认的显示等级，显示所有信息
os.environ["TF_CPP_MIN_LOG_LEVEL"]='2'
# 只显示 warning 和 Error  
os.environ["TF_CPP_MIN_LOG_LEVEL"]='3'
# 只显示 Error