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; модель поддерживает API Predict. Чтобы использовать API Classify, Regress или MultiInference, см. раздел 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 | Поддержка устаревшего формата для операции или группы операций, выполняемых после операции восстановления при загрузке. |
© 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/api_docs/python/tf/compat/v1/saved_model/simple_save