停止 Tensorflow 打印到控制台

Question

我一直使用tensorflow没有问题，直到我添加了以下代码行：

log_dir = os.path.join("logs",

                       "fit",

                       datetime.datetime.now().strftime("%Y%m%d-%H%M%S"))

tensorboard_callback = TensorBoard(log_dir)

运行此命令后，我会在控制台上打印大量信息。我尝试查看 tf.keras.callbacks.TensorBoard 文档，看看是否可以减少冗长，但我没有看到任何选项。

从各种stackoverflow 答案中，我也尝试设置tfdown 的详细程度，但无济于事：

tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.ERROR)

tf.get_logger().setLevel('ERROR')

tf.autograph.set_verbosity(3)

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

我有以下规格：

Python = 3.8

Tensorflow = 2.3.1

Cuda Toolkit = 10.1 

cuDNN = 7.6.4

GPU=Nvidia RTX2060

打印到控制台的信息都是I消息

Answer 1

您可以使用 禁用调试日志os.environ。import osos.environ['TF_CPP_MIN_LOG_LEVEL'] = '3' import tensorflow as tf可能的值如下：0 = all messages are logged (default behavior)1 = INFO messages are not printed2 = INFO and WARNING messages are not printed3 = INFO, WARNING, and ERROR messages are not printed