Spec-Zone.ru › TensorFlow 1.15

tf.contrib.tpu.export_estimator_savedmodel

Экспорт обученной модели для вывода на TPU.

tf.contrib.tpu.export_estimator_savedmodel(
    estimator, export_dir_base, serving_input_receiver_fn, assets_extra=None,
    as_text=False, checkpoint_path=None
)
Аргументы
estimator Estimator, с помощью которой была обучена модель.
export_dir_base Строка, содержащая директорию, в которой будут созданы поддиректории с отметкой времени, содержащие экспортированные SavedModel.
serving_input_receiver_fn Функция, которая не принимает аргументов и возвращает ServingInputReceiver или TensorServingInputReceiver.
assets_extra Словарь, определяющий, как заполнить директорию assets.extra экспортированного SavedModel, или None, если дополнительные ресурсы не требуются.
as_text Флаг, указывающий, следует ли записать протокол SavedModel в текстовом формате.
checkpoint_path Путь к контрольной точке для экспорта. Если None (по умолчанию), выбирается самая последняя контрольная точка, найденная в каталоге модели.
Возвращаемое значение
Путь к экспортированной директории.

© 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/contrib/tpu/export_estimator_savedmodel

Spec-Zone.ru

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