tf.saved_model.simple_save
Функция для удобного создания SavedModel, подходящей для обслуживания. (устаревшая)
tf.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 и поддерживает API прогнозирования. Для использования API классификации, регрессии или MultiInference, используйте либо tf.Estimator, либо более низкоуровневые API SavedModel.
- Некоторые операторы 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 | Поддержка устаревшего функционала для выполнения оператора или группы операторов после оператора восстановления при загрузке. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/saved_model/simple_save