Spec-Zone.ru › TensorFlow

tf.compat.v1.saved_model.simple_save

Вспомогательная функция для создания SavedModel, подходящей для обслуживания. (устарела)

tf.compat.v1.saved_model.simple_save(
    session, export_dir, inputs, outputs, legacy_init_op=None
)

Используется в блокнотах

Используется в руководстве
  • Перенос кода TFLite на TF2
  • Перенос рабочего процесса SavedModel
Устарело: ЭТА ФУНКЦИЯ УСТАРЕЛА. Она будет удалена в будущей версии. Инструкции по обновлению: этот API разработан для TensorFlow v1. Инструкции по переносу кода на TensorFlow v2 см. на странице https://www.tensorflow.org/guide/migrate.

Во многих распространённых случаях сохранение моделей для обслуживания сводится к следующему:

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

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API