tf.compat.v1.saved_model.simple_save
Функция для удобного создания SavedModel, подходящего для предоставления сервиса. (устаревшая)
tf.compat.v1.saved_model.simple_save(
session, export_dir, inputs, outputs, legacy_init_op=None
)
Во многих распространённых случаях сохранение моделей для предоставления сервиса будет таким же простым, как:
simple_save(session,
export_dir,
inputs={"x": x, "y": y},
outputs={"z": z})
Хотя во многих случаях нет необходимости понимать все способы конфигурации SavedModel, этот метод имеет несколько практических последствий:
- Он будет рассматриваться как граф для инференции/предоставления сервиса (т.е. используется метка
saved_model.SERVING) - SavedModel будет загружаться в TensorFlow Serving и поддерживает Predict API. Для использования API Classify, Regress или MultiInference, пожалуйста, используйте либо tf.Estimator, либо более низкоуровневые SavedModel API.
- Некоторые операции TensorFlow зависят от информации на диске или другой информации, называемой "активами". Обычно они обрабатываются автоматически путём добавления активов в
GraphKeys.ASSET_FILEPATHSколлекцию. Экспортируются только активы в этой коллекции; если вам требуется более настраиваемое поведение, вам нужно использовать SavedModelBuilder.
Дополнительную информацию о SavedModel и сигнатурах можно найти здесь: https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/saved_model/README.md
| Аргументы | |
|---|---|
session | TensorFlow сессия, из которой необходимо сохранить метаграф и переменные. |
export_dir | Путь, по которому будет сохранён SavedModel. |
inputs | Словарь, отображающий строковые имена входящих данных на тензоры. Эти данные добавляются в SignatureDef в качестве входящих данных. |
outputs | Словарь, отображающий строковые имена выходных данных на тензоры. Эти данные добавляются в SignatureDef в качестве выходных данных. |
legacy_init_op | Поддержка устаревшего варианта для выполнения операции или группы операций после операции восстановления при загрузке. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/compat/v1/saved_model/simple_save