用下面代码设置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
通过设置环境变量TF_CPP_MIN_LOG_LEVEL为3,并使用tf.get_logger().setLevel(ERROR),可以确保在TensorFlow2.3中屏蔽所有警告信息,只显示错误级别日志。
2457

被折叠的 条评论
为什么被折叠?



