tensorflow keras 关闭打印warning

skyecs

于 2023-06-30 15:15:03 发布

阅读量564

点赞数

文章标签： tensorflow keras 人工智能

本文链接：https://blog.csdn.net/skyecs/article/details/131477266

版权

用下面代码设置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