我正在使用以下代码保存模型。
callbacks = [keras.callbacks.ModelCheckpoint('DeeplabV3Plus', save_best_only=True),
keras.callbacks.EarlyStopping(patience=30, restore_best_weights=True)]
使用h5格式时,不会显示如下。
如何才能不显示这些消息?
我必须保存没有 h5 格式的模型,因为 h5 格式使我保存时出错。
但是这些消息来了。不过,模型保存和加载就可以了。
我只是不想显示这些消息。
根据@Sagar,答案如下。
要阻止 TensorFlow 在保存模型时显示信息性消息,您可以更改其日志记录设置以仅显示错误等重要消息。 ************使用
tf.get_logger().setLevel(logging.ERROR)
仅显示错误并隐藏信息性消息。 ************使用 os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
也可以隐藏警告