Я пытаюсь сохранить и загрузить модели Keras и/или TensorFlow и подхожу к проблеме, следуя документации Keras и руководству по TensorFlow. В VS Code на моем ПК как собственный формат .keras, так и формат SavedModel, описанный в руководстве по TensorFlow, возвращают ошибки. Однако следующий почти минимальный пример с использованием устаревшего формата .h5 завершается без ошибок (этот пример адаптирован из документации Keras).
import keras
model = keras.Sequential(
[
keras.layers.Dense(5, input_shape=(3,)),
keras.layers.Softmax(),
],
)
model.save('model.h5')
loaded_model = keras.saving.load_model('model.h5')
И тем не менее, он также отображает два предупреждения, которые раздражают при многократном запуске модели:
ПРЕДУПРЕЖДЕНИЕ:absl:Вы сохраняете вашу модель в виде файла HDF5 с помощью model.save() или keras.saving.save_model(model). Этот формат файла считается устаревшим. Вместо этого мы рекомендуем использовать собственный формат Keras, например. model.save('my_model.keras') или keras.saving.save_model(model, 'my_model.keras').
< strong>ВНИМАНИЕ:absl:В файле сохранения не найдена конфигурация обучения, поэтому модель не скомпилирована. Скомпилируйте его вручную.
Первый соответствует методу .save, а второй — методу .load_model.
Как можно подавить эти предупреждения, чтобы они не появлялись при каждом запуске?
Для справки я попробовал добавить следующее без успех.
import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
import tensorflow as tf
tf.get_logger().setLevel('ERROR')
import warnings
warnings.filterwarnings("ignore")
import tensorflow as tf
tf.keras.config.disable_interactive_logging()
Подробнее здесь: https://stackoverflow.com/questions/782 ... ide-the-wa
При сохранении и загрузке моделей Keras или TensorFlow: как отключить/скрыть предупреждения, которые появляются при испо ⇐ Python
-
- Похожие темы
- Ответы
- Просмотры
- Последнее сообщение