如何不显示INFO:tensorflow:Assets写入?

问题描述 投票:0回答:1

我正在使用以下代码保存模型。

callbacks = [keras.callbacks.ModelCheckpoint('DeeplabV3Plus', save_best_only=True),
            keras.callbacks.EarlyStopping(patience=30, restore_best_weights=True)]

使用h5格式时,不会显示如下。

enter image description here

如何才能不显示这些消息?

我必须保存没有 h5 格式的模型,因为 h5 格式使我保存时出错。

但是这些消息来了。不过,模型保存和加载就可以了。

我只是不想显示这些消息。

tensorflow keras
1个回答
0
投票

根据@Sagar,答案如下。

要阻止 TensorFlow 在保存模型时显示信息性消息,您可以更改其日志记录设置以仅显示错误等重要消息。 ************使用

tf.get_logger().setLevel(logging.ERROR)
仅显示错误并隐藏信息性消息。 ************使用
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
也可以隐藏警告

© www.soinside.com 2019 - 2024. All rights reserved.